这道题单眼一看很简单,不就是求三个四位数,满足每一位都相等或者都不想等就可以吗,但是仔细想想,能想到的好像只有三成循环,能找的也只是在三层循环的基础上面去找优化方法,但出题方就是算准了,你用三层就是卡你,2s也过不了,我苦苦思索,多方求佬,终于得出了一个相对比较好理解的思路和方法:
cnt标记数组,标记和要寻找的第三个数的每一位的相等或者不相等的数的个数的数组
思路就是对每个数的每一位做标记,也就是当我们拿到某个数A的时候,那么相对于另外一个数B来说,A要么和B的某一位相等,要么不相等,这就有16种可能,这样的话,我们就能根据给出的所有的数A建立一个cnt的标记数组,用来查询和某个数B的某一位相等或者不相等的数的个数有多少。
find搜索函数 (能求出来能和某两个数组成三元组的数的个数)
接下来就是一个find函数,用来查询和某一个数的某一位相等还是不想等的元素的个数,如果相等,就找全部都相等的相加,如果不相等,就加上所有含当前位(与与要寻找的数的当前位不相等的数的个数)的数,然后减去和a当前位相等的,和b当前位相等的数的个数,也就是符合题目要求的三个数的某一位都不相等的数的个数。
两层循环判断第三个数(先取任意两个数)
这里的a和b是我们两层循环得到的能组成三元组的某两个数。
这样的话 ,我们就能通过标记数组通过某个数位的相等与否直接得到第三个数的所有取值可能,从而把三层循环降为两层。
#include "iostream"
using namespace std;
int cnt[11][11][11][11];
inline void sl(int *a){ //与某一个数字相等的位和不相等的位的个数都列出来
cnt[a[0]][a[1]][a[2]][a[3]] ++ ; //1111
cnt[a[0]][a[1]][a[2]][0] ++; //1110
cnt[a[0]][a[1]][0][a[3]]++ ; // 1101
cnt[a[0]][a[1]][0][0] ++; // 1100
cnt[a[0]][0][a[2]][a[3]] ++ ; //1011
cnt[a[0]][0][a[2]][0] ++; //1010
cnt[a[0]][0][0][a[3]] ++ ; //1001
cnt[a[0]][0][0][0] ++; //1000
cnt[0][a[1]][a[2]][a[3]] ++ ; //0111
cnt[0][a[1]][a[2]][0] ++; //0110
cnt[0][a[1]][0][a[3]] ++ ; //0101
cnt[0][a[1]][0][0] ++; // 0100
cnt[0][0][a[2]][a[3]] ++ ;// 0011
cnt[0][0][a[2]][0] ++; //0010
cnt[0][0][0][a[3]] ++ ; //0001
cnt[0][0][0][0] ++; //0000
}
long long find(int now, int *a, int *b, int *c){
long long sum = 0;
if(now == 4) sum += cnt[c[0]][c[1]][c[2]][c[3]];
else if(a[now] == b[now]){
c[now] = a[now];
sum += find(now +1, a, b, c);
}
else {
c[now] = 0;
sum += find(now + 1, a, b,c); //先加上所有和当前位不相等的数的数量
c[now] = a[now]; //减去当前位和a的当前位相等的数的数量
sum -= find(now + 1, a, b, c);
c[now] = b[now]; //减去当前位和b的当前位相等的数的数量
sum -= find(now + 1, a, b, c);
}
return sum;
}
int main(){
ios::sync_with_stdio(false),cin.tie(0), cout.tie(0);
int t;
cin>>t;
long long ans = 0;
int num[3000][7];
for(int i = 0; i < t; i++){
string s;
cin>>s;
for(int j = 0; j < 4; j ++){
num[i][j] = s[j] - '0';
}
sl(num[i]);
}
int ann[5];
for(int i = 0; i < t; i ++){
for(int j = i + 1; j < t; j ++){
ans += find(0, num[i], num[j], ann);
}
}
cout<<ans/3<<endl;
return 0;
}