[codeforces] 633C Spy Syndrome 2

原题

Trie树+dp

首先，我们可以简单的想到一种dp方式，就是如果这一段可以匹配并且可以与前一段接上，那么更新dp[i]为当前字符串的编号，然后倒推就可以得到答案。

但是，显然我们不能O(m)比较，那么怎么办呢？

这时候就可以体现Trie树的意义了，我们在爬Trie树的过程中就可以完成判断所有合法的字符串。当我们可以更新一个dp的时候就break就好了。

#include<cstdio>
#include<cstring>
#include<vector>
#define M 1001000
#define N 10010
using namespace std;
int n,m,dp[M],l[M],cnt=2,now,ll,pos[M];
char s[M],t[M];
vector <int> v;
struct hhh
{
    int son[26],is;
    hhh()
	{
	    is=0;
	    memset(son,0,sizeof(son));
	}
}edge[M];

void add(int x)
{
    now=1;
    int q;
    for (int i=ll+1;i<=ll+l[x];i++)
    {
	if (t[i]<'a') q='a'-'A';else q=0;
	if (edge[now].son[t[i]-'a'+q]) now=edge[now].son[t[i]-'a'+q];
	else
	{
	    edge[now].son[t[i]-'a'+q]=cnt++;
	    now=edge[now].son[t[i]-'a'+q];
	}
    }
    edge[now].is=x;
}

int main()
{
    scanf("%d",&n);
    scanf("%s",s+1);
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
	scanf("%s",t+ll+1);
	pos[i]=ll+1;
	l[i]=strlen(t+ll+1);
	add(i);
	ll+=l[i];
    }
    dp[0]=1;
    for (int i=1;i<=n;i++)
    {
	now=1;
	for (int j=1;j<=i && now;j++)
	{
	    now=edge[now].son[s[i-j+1]-'a'];
	    if (edge[now].is && dp[i-j])
	    {
		dp[i]=edge[now].is;
		break;
	    }
	}
    }
    for (int i=n;i>=1;i-=l[dp[i]])
	v.push_back(dp[i]);
    int ss=v.size();
    for (int i=ss-1;i>=0;i--)
    {
	for (int j=1;j<=l[v[i]];j++)
	    putchar(t[pos[v[i]]+j-1]);
	putchar(' ');
    }
    return 0;
}