Codeforces 633 C Spy Syndrome 2 字典树

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N=1000000+5;
char s[10005];
char str[100005][1005];
int len[100005];
int trie[N][26],mk[N];
int ans[10005],tot,cnt,m,n;
void add(int x)
{
    int now=0;
    for(int i=len[x]-1; i>=0; --i)
    {
        char tmp=str[x][i];
        if(tmp<‘a‘||tmp>‘z‘)
            tmp+=‘a‘-‘A‘;
        int p=tmp-‘a‘;
        if(!trie[now][p])
            trie[now][p]=++cnt;
        now=trie[now][p];
    }
    mk[now]=x;
}
void solve(int pos)
{
    if(pos==n)
    {
        for(int i=1; i<tot; ++i)
            printf("%s ",str[ans[i]]);
        printf("%s\n",str[ans[tot]]);
       exit(0);
    }
    int now=0;
    for(int i=pos+1; i<=n; ++i)
    {
        int p=s[i]-‘a‘;
        now=trie[now][p];
        if(!now)break;
        if(mk[now])
        {
            ans[++tot]=mk[now];
            solve(pos+len[mk[now]]);
            --tot;
        }
    }
}
int main()
{
    scanf("%d%s",&n,s+1);
//    for(int i=1; i<=n/2; ++i)
//        swap(s[i],s[n-i+1]);
    scanf("%d",&m);
    for(int i=1; i<=m; ++i)
    {
        scanf("%s",str[i]);
        len[i]=strlen(str[i]);
        add(i);
    }
    solve(0);
    return 0;
}