Codeforces 633C Spy Syndrome 2 【Trie树】+【DFS】

<题目链接>

题目大意：
给定一个只有小写字母组成的目标串和m个模式串(里面可能有大写字母)，记目标串反过来后的串为S，让你从m个模式串中选出若干个组成S串(不区分大小写)。输出任意一种方案。

解题分析：
将所有单词倒着建好Trie树后(字母忽略大小写)，直接在Trie树上跑DFS，记录下所有符合条件的单词序号，然后输出即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N =1e6+10;
const int M =1e5+10;
int n,m;
char s[M],word[M][1005];
int trie[N][27],mark[N],ans[M],cnt,tot;
void Insert(char *str,int ord){
    int now=0,len=strlen(str);
    for(int i=len-1;i>=0;i--){    //对这些单词进行逆序插入
        int to;
        if(str[i]>'Z')to=str[i]-'a';    //忽略单词的大小写
        else to=str[i]-'A';
        if(!trie[now][to])
            trie[now][to]=++tot;
        now=trie[now][to];
    }
    mark[now]=ord;    //记录下该单词的编号
}
bool dfs(int loc){
    if(s[loc]=='\0'){    //搜索结束
        for(int i=0;i<cnt;i++){
            printf("%s%s",word[ans[i]],i==cnt-1?"\n":" ");
        }
        return true;
    }
    int now=0;
    for(int i=loc;i<n;i++){    
        int to=s[i]-'a';
        if(!trie[now][to])break;   //如果不符合，说明这种搜索的情况不符，直接结束跳出
        now=trie[now][to];
        if(mark[now]){    //如果搜到了单词库中完整的单词
            ans[cnt++]=mark[now];   //记录下该单词的序号
            if(dfs(i+1))return true;   //继续向下搜索
            cnt--;    //如果之前的情况都不符合，这里进行回溯，继续对下一个字符进行搜索
        }
    }
    return false;
}
int main(){
    tot=cnt=0;
    scanf("%d%s",&n,s);
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",word[i]);
        Insert(word[i],i);
    }
    dfs(0);
    return 0;
}

 

 

2018-11-03