【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]);
}