USACO 1.3 Combination Lock

题意概述：
农夫约翰的奶牛不停地从他的农场中逃出来，导致了很多损害。为了防止它们再逃出来，他买了一只很大的号码锁以防止奶牛们打开牧场的门。
农夫约翰知道他的奶牛很聪明，所以他希望确保它们不会在简单地试了很多不同的号码组合之后就能轻易开锁。锁上有三个转盘，每个上面有数字1..N (1 <= N <= 100)，因为转盘是圆的，所以1和N是相邻的。有两种能开锁的号码组合，一种是农夫约翰设定的，还有一种“预设”号码组合是锁匠设定的。但是，锁有一定的容错性，所以，在每个转盘上的数字都与一个合法的号码组合中相应的数字相距两个位置以内时，锁也会打开。
比如说，如果农夫约翰的号码组合是(1,2,3)，预设号码组合是(4,5,6)，在转盘被设定为(1,4,5)（因为这和农夫约翰的号码组合足够接近）或(2,4,8)（因为这和预设号码组合足够接近）。注意，(1,5,6)并不会打开锁，因为它与任一号码组合都不够接近。
给出农夫约翰的号码组合和预设号码组合，请计算能够开锁的不同的号码组合的数目。号码是有序的，所以(1,2,3)与(3,2,1)不同。
解题思路：
分两种情况来解决，如果n小于5，那么一定会是重复的，也就是说这个时候只有n^3种情况。否则，则采取从不重复的情况下的总数250中去掉重复的情况的个数即可，去的方法是每个位置上都有一个权值，分别计算出来后乘在一起即可。
题解代码：

#include<bits/stdc++.h>
using namespace std;
int n;
int extra(int x,int a){
    int z1;
    int offset=5;
    for(int k=0;k<=4;k++){
        if((x+k)%n==a%n||(x-k)%n==a%n||(a+k)%n==x%n||(a-k)%n==x%n){
            z1=offset;
            break;
        }
        offset--;
        if(k==4)z1=0;
    }
    return z1;
} 
int main(){
    ifstream fin("combo.in");
    ofstream fout("combo.out");
    int res=250;
    fin>>n;
    int a,b,c,x,y,z;
    fin>>a>>b>>c>>x>>y>>z;
    int z1,z2,z3;
    if(n>=5){
        z1=extra(x,a);
        z2=extra(y,b);
        z3=extra(z,c);
        res=res-z1*z2*z3;
    }
    else
        res=n*n*n;
    fout<<res<<endl;
}