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+;
 const int M =1e5+;
 int n,m;
 char s[M],word[M][];
 int trie[N][],mark[N],ans[M],cnt,tot;
 void Insert(char *str,int ord){
     int now=,len=strlen(str);
     for(int i=len-;i>=;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=;i<cnt;i++){
             printf("%s%s",word[ans[i]],i==cnt-?"\n":" ");
         }
         return true;
     }
     int now=;
     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+))return true;   //继续向下搜索
             cnt--;    //如果之前的情况都不符合，这里进行回溯，继续对下一个字符进行搜索
         }
     }
     return false;
 }
 int main(){
     tot=cnt=;
     scanf("%d%s",&n,s);
     scanf("%d",&m);
     for(int i=;i<=m;i++){
         scanf("%s",word[i]);
         Insert(word[i],i);
     }
     dfs();
     return ;
 }

2018-11-03