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-1+N)%N+1; }
bool inRange(int x,int y) { return x==y || add(x,1)==y || add(x,2)==y || add(x,-1)==y || add(x,-2)==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=0;
    Key t;
    for(t.x=1;t.x<=N;++t.x)
        for(t.y=1;t.y<=N;++t.y)
            for(t.z=1;t.z<=N;++t.z) s+=(t==A||t==B);
    out<<s<<endl;

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