单词 7.29 T2

时间限制: 5 Sec 内存限制: 512 MB
题目描述

输入

样例输入
4 3
xyz
xyz
zzx
xzz
样例输出
2
1
2
1

开始接受zyh的暑假训练，不过从7.26开始状态一直不怎么样，机房里闷热的环境和喧嚣的打隔膜声，也让我感觉写题很难受。所以今天就给自己放了一个假。睡完两个小时的午觉，总算有力气开始填坑了。

std是一个dp的做法（看也看不懂），所以我就写了一个trie树，方法其实很简单，每次先遍历一遍trie树，再把字符串添加进去就行了。

#include
#include
#include
#include
#define ll long long
using namespace std;
int n,m,len,tot;
char s[100005];
ll h[100005];
struct ty
{
    int Next[3],cnt;
    void init()
    {
        memset(Next,-1,sizeof(Next));
        cnt=0;
    }
}p[100005];
void update(char *s) 
{
    int len=strlen(s);
    int x=0;
    for(int i=0;iint j=s[i]-'x';
        if(p[x].Next[j]==-1) 
        {
            tot++;
            p[x].Next[j]=tot;
            p[tot].init();
        }
        x=p[x].Next[j];
        p[x].cnt++;
    }
}
void solve(int x,int k,int sum) //遍历trie树 
{
    if(k==len) 
    {
        h[sum]=h[sum]+p[x].cnt;
        return;
    }
    for(int i=0;i<3;i++) 
    if(p[x].Next[i]!=-1) solve(p[x].Next[i],k+1,(sum+((s[k]-'x')==i)));
}       
int main()
{
    p[0].init();
    cin>>n>>m;
    for(int i=1;i<=n;i++) 
    {
        scanf("%s",s);
        len=strlen(s);
        solve(0,0,0);
        update(s);
    }
    for(int i=0;i<=m;i++) printf("%lld\n",h[i]);
    return 0;
}