题目大意
给出若干个四个字符组成的字符串。
求有多少对仅有m位不同。
DP
设f[i]表示恰有i位不同的对数。
只有i位不同,意味着如果无视其中的i位,总有一种方案使得两个字符串完全一致。
于是暴力枚举无视哪i位,然后将无视后的字符集排序统计相同对数和记为t。
注意其中可能包括恰有j位不同的对(
j<i
)所以要减掉。
也就是
f[i]=t−∑i−1j=0Ci−j4−j
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=50000+10;
char s[maxn][4];
int id[maxn];
bool bz[5];
int c[5][5];
int f[5];
int i,j,k,l,t,n,m;
bool cmp(int x,int y){
int i;
fo(i,0,3)
if (bz[i]&&s[x][i]!=s[y][i]) return 0;
return 1;
}
bool cmp2(int x,int y){
int i;
fo(i,0,3)
if (bz[i]){
if (s[x][i]<s[y][i]) return 1;
else if (s[x][i]>s[y][i]) return 0;
}
return 0;
}
bool czy(int a,int b){
return cmp2(a,b);
}
void dfs(int x,int y){
if (4-x<y) return;
if (x==4){
int i,j=1,k,l;
fo(i,1,n) id[i]=i;
sort(id+1,id+n+1,czy);
fo(i,2,n+1)
if (i>n||!cmp(id[i],id[j])){
k=i-j;l=i-j-1;
if (k%2) l/=2;else k/=2;
t+=k*l;
j=i;
}
return;
}
if (y){
bz[x]=1;
dfs(x+1,y-1);
}
bz[x]=0;
dfs(x+1,y);
}
int main(){
fo(i,1,4) c[0][i]=0;
c[0][0]=1;
fo(i,1,4){
c[i][0]=1;
fo(j,1,4)
c[i][j]=c[i-1][j]+c[i-1][j-1];
}
scanf("%d%d",&n,&m);
fo(i,1,n) scanf("%s",s[i]);
fo(i,0,m){
t=0;
dfs(0,4-i);
fo(j,0,i-1) t-=f[j]*c[4-j][i-j];
f[i]=t;
}
printf("%d\n",f[m]);
}