poj 1816 (Trie + dfs)

题目链接：http://poj.org/problem?id=1816

思路：建好一颗Trie树，由于给定的模式串可能会重复，在原来定义的结构体中需要增加一个vector用来记录那些以该节点为结尾的字符串的序号，然后就是匹配的过程了，需要注意的是，对于‘？'和'*',每一次都是可以匹配的，并且对于'*'，还得枚举所需要匹配的长度（从0开始）。由于最后的答案可能会有重复，一开始我没有判断时候有重复的，WA了好多次=.=!!.

  1 #include <iostream>
  2 #include <cstdio>
  3 #include <cstring>
  4 #include <algorithm>
  5 #include <vector>
  6 #define REP(i, a, b) for (int i = (a); i <= (b); ++i)
  7 using namespace std;
  8 
  9 const int MAN_N = (100000 + 100);
 10 struct Trie {
 11     int index;
 12     Trie *next[28];
 13     vector<int > ID; //可能会重复
 14     Trie()
 15     {
 16         index = -1;
 17         memset(next, 0, sizeof(next));
 18         ID.clear();
 19     }
 20 };
 21 
 22 Trie *root;
 23 int getID(char ch)
 24 {
 25     if (ch >= 'a' && ch <= 'z') return ch - 'a';
 26     else if (ch == '?') return 26;
 27     return 27;
 28 }
 29 void Insert(char *str, int pos)
 30 {
 31     int len = strlen(str);
 32     Trie *p = root;
 33     REP(i, 0, len - 1) {
 34         int id = getID(str[i]);
 35         if (p->next[id] == NULL) {
 36             p->next[id] = new Trie();
 37         }
 38         p = p->next[id];
 39     }
 40     p->index = pos;
 41     p->ID.push_back(pos);
 42 }
 43 int N, M;
 44 char str[22];
 45 vector<int > ans;
 46 
 47 void Check(Trie *&p)
 48 {
 49     if (p->next[27]) {
 50         if (p->next[27]->index != -1) {
 51             REP(i, 0, (int)p->next[27]->ID.size()-1) ans.push_back(p->next[27]->ID[i]);
 52         }
 53         Check(p->next[27]);
 54     }
 55 }
 56 
 57 
 58 void dfs(Trie *&p, int pos)
 59 {
 60     if (pos == N) {
 61         if (p->index != -1) {
 62             REP(i, 0, (int)p->ID.size()-1) {
 63                 ans.push_back(p->ID[i]);
 64             }
 65         }
 66         Check(p);
 67     } else {
 68         if (p->next[getID(str[pos])]) dfs(p->next[getID(str[pos])], pos + 1);
 69         if (p->next[26]) dfs(p->next[26], pos + 1);
 70         if (p->next[27]) {
 71             REP(i, pos, N) dfs(p->next[27], i);
 72         }
 73     }
 74 }
 75 
 76 
 77 int main()
 78 {
 79     cin >> N >> M;
 80     root = new Trie();
 81     REP(i, 0, N - 1) {
 82         scanf("%s", str);
 83         Insert(str, i);
 84     }
 85     REP(i, 0, M - 1) {
 86         scanf("%s", str);
 87         N = strlen(str);
 88         dfs(root, 0);
 89         if ((int)ans.size() > 0) {
 90             sort(ans.begin(), ans.end());
 91             printf("%d", ans[0]);
 92             REP(i, 1, (int)ans.size()-1) {
 93                 if (ans[i] != ans[i - 1]) printf(" %d", ans[i]);
 94             }
 95             puts("");
 96         } else
 97             puts("Not match");
 98         ans.clear();
 99     }
100     return 0;
101 }

View Code