USACO Section1.3 Combination Lock 解题报告

　　　　combo解题报告 —— icedream61 博客园（转载请注明出处）
------------------------------------------------------------------------------------------------------------------------------------------------
【题目】
　　一个三位密码锁，每位可能取值1~N。有两个解锁密码，都会给出。
　　若尝试的密码与某个解锁密码相等或相近（每位值都至多差2），均可解开锁。
　　例如：(1,2,3)(4,5,6)是两个解锁密码，那么(1,N,5)(2,4,8)均可解锁，而(1,5,6)则不可。
【数据范围】
　　1<=N<=100
【输入样例】
　　50
　　1 2 3
　　5 6 7
【输出样例】
　　249
------------------------------------------------------------------------------------------------------------------------------------------------
【分析】
　　直接枚举。
　　需要注意，两个密码相等的判断容易出错（需要%N的情况下，下标并非从0开始；差2及以内均相等，此差2可能与%N同时发生）。
------------------------------------------------------------------------------------------------------------------------------------------------
【总结】
　　一遍AC。
　　顺带用了下重载运算符与类内置函数的语法，有个小问题存疑：如果我把代码中add和inRange两个函数放到Key类里面，那么operator==里面便无法调用这两个函数。如有人懂，望不吝赐教～

------------------------------------------------------------------------------------------------------------------------------------------------

【代码】

 /*
 ID: icedrea1
 PROB: combo
 LANG: C++
 */

 #include <iostream>
 #include <fstream>
 using namespace std;

 int N;

 int add(int x,int y) { return (x+y-+N)%N+; }
 bool inRange(int x,int y) { return x==y || add(x,)==y || add(x,)==y || add(x,-)==y || add(x,-)==y; }

 struct Key
 {
     int x,y,z;
     friend bool operator==(Key a,Key b) { return inRange(a.x,b.x) && inRange(a.y,b.y) && inRange(a.z,b.z); }
 };

 Key A,B;

 int main()
 {
     ifstream in("combo.in");
     ofstream out("combo.out");

     in>>N;
     in>>A.x>>A.y>>A.z;
     in>>B.x>>B.y>>B.z;

     int s=;
     Key t;
     for(t.x=;t.x<=N;++t.x)
         for(t.y=;t.y<=N;++t.y)
             for(t.z=;t.z<=N;++t.z) s+=(t==A||t==B);
     out<<s<<endl;

     in.close();
     out.close();
     return ;
 }