[USACO1.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。
第二行：三个以空格分隔的整数，为农夫约翰的号码组合。
第三行：三个以空格分隔的整数，为预设号码组合（可能与农夫约翰的号码组合相同）。

输出格式：

第一行：所有不同的能够开锁的号码组合的总数。

输入输出样例

输入样例#1：

50
1 2 3
5 6 7

输出样例#1：

249

说明

每个转盘的标号是1..50。农夫约翰的号码组合是(1,2,3)，预设号码组合是(5,6,7)。

题目翻译来自NOCOW。

题解

其实就是一个简单的排列组合问题，首先我们知道最好的情况是预设号码和农夫号码完全不同的时候。可以求出来这是一个常量，即5×5×5×2=250。所以我们只要求出有多少个重复的号码组合就行了。（当然你也可以暴力直接三重循环毕竟n的范围很小）找重复的话直接用个标记数组就可以了，当然如果数据大的话可能就要用hash了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,ans;
int key[105][105][105];
int a,b,c;
int x,y,z;
int mod(int x)
{
    return (x+n)%n;
}//因为最后一位和第一位相连
int main()
{
    scanf("%d",&n);
    if(n<=5)
    {
        printf("%d\n",n*n*n);
        return 0;
    }//特判因为n小于5的时候还不到250种情况
    scanf("%d%d%d%d%d%d",&a,&b,&c,&x,&y,&z);
    for(int i=a-2;i<=a+2;++i)
        for(int j=b-2;j<=b+2;++j)
            for(int k=c-2;k<=c+2;++k)
                key[mod(i)][mod(j)][mod(k)]=1;
    for(int i=x-2;i<=x+2;++i)
        for(int j=y-2;j<=y+2;++j)
            for(int k=z-2;k<=z+2;++k)
                if(key[mod(i)][mod(j)][mod(k)]==1) ans++;
    printf("%d\n",250-ans);
    return 0;
}