【GDOI模拟】习用之语

Description

Solution

中华文化，博大精深！

一看到这题，第一眼的反应就是字典树。

但是仔细一想，好像要开一两二三四….好多颗字典树，好麻烦啊……

想到代码量这么长，就不想打了。

然后往短里想，好像判断某几位的时候可以用排序来做。

通过c++的优势，开始打排序的判断函数，一两二三四……好多个啊……

又不想打了。

最后才发现一开始一点都不机智。

我用一个dfs枚举要判断哪几位，然后把别的位删掉，在拍个序就好了TAT。

然后我们先求的选了i为比较，一共有f[i]对相同的(不判重),上面就是求这个的。

哪儿我们设答案为g（是一个数组）,g[i]表示选了i位，只有i位不同的对数。

其实可以用容斥搞出来的，但是数据非常的小，手动推一推就好了。

首先g[0]=f[4]（这个很显然，因为只有四位是相同的就是有四位是相同的(f[4])）。

g[1]=f[3]-4*g[0]（仅为1个不同（即仅为3个相同），在仅为0个不同（即全部相同）里算了4次(就是在四个数中选3个一共有四种方案)，这4次都要去掉，后面同理）。

g[2]=f[2]-g[1]*3-g[0]*6（在有3个相同里选两个一共有三种方案，在有4个相同里选两个一共有六种方案）

g[3]=f[1]-g[2]*2-g[1]*3-g[0]*4（在有2个相同里选1个一共有2种方案，在有3个相同里选1个一共有3种方案，（在有3个相同里选两个一共有三种方案，在有4个相同里选1个一共有4种方案）

g[4]=n*(n-1)/2-g[0]-g[1]-g[2]-g[3]（这里就没必要继续往下推了，直接用总的减去不合法的就好了）

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=50007;
int i,j,k,l,t,n,m,ans,ans1,r,root,num,b[5];
int f[5],g[5];
char s[maxn][5],st[maxn][5];
struct node{
char s[5];
}a[maxn];
bool cmp(node x,node y){
int i;
fo(i,1,4){
if(x.s[i]<y.s[i])return 1;
else if(x.s[i]>y.s[i])return 0;
}
return 0;
}
void dfs(int x,int y){
int i,j,k;
if(y>=1){
fo(i,1,n){
fo(j,1,y){
a[i].s[j]=s[i][b[j]];
}
}
sort(a+1,a+1+n,cmp);
int l=r=1;
a[n+1].s[1]='#';
fo(i,2,n+1){
bool bz=1;
fo(j,1,y){
if(a[i-1].s[j]!=a[i].s[j]){
bz=0;
break;
}
}
if(bz)r=i;
else{
f[y]+=(r-l+1)*(r-l)/2;
l=r=i;
}
}
}
if(y==4)return;
fo(i,x+1,4){
b[y+1]=i;
dfs(i,y+1);

}
}
int main(){
scanf("%d%d",&n,&m);
fo(i,1,n){
scanf("%s",s[i]+1);
}
dfs(0,0);
g[0]=f[4];g[1]=f[3]-g[0]*4;
g[2]=f[2]-g[1]*3-g[0]*6;
g[3]=f[1]-g[2]*2-g[1]*3-g[0]*4;
if(n%2==0)g[4]=n/2*(n-1)-g[0]-g[1]-g[2]-g[3];
else g[4]=(n-1)/2*n-g[0]-g[1]-g[2]-g[3];
printf("%d\n",g[m]);
}