AcWing1351. 密码锁（容斥原理）

题目：1351. 密码锁

题解：当n<=5时，所以方案都行，输出n^3即可。当N>5时，555+555-两者公共的。两者公共的就是看：第一位里面公共的数✖第二位里面公共的数✖第三位里面公共的数。

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef long long LL;
typedef pair<int ,int> PII;
const int N=1e5+10;
const int mod=100000007;
set<int > st[5];
int main(){
    int n;
    cin>>n;
    int a[5];
    cin>>a[0]>>a[1]>>a[2];
    for(int i=0;i<3;i++){
        for(int j=-2;j<=2;j++){
            st[i].insert((a[i]+j+n)%n);//默认将n看成0
        }
    }
    cin>>a[0]>>a[1]>>a[2];
    int sum=1;
    for(int i=0;i<3;i++){
        int ans=0;
        for(int j=-2;j<=2;j++){
            if(st[i].find((a[i]+j+n)%n)!=st[i].end()) ans++;
        }
        sum*=ans;
    }
    if(n<=5) cout<<n*n*n;
    else{
        cout<<5*5*5*2-sum;
    }
    return 0;
}