poj 3630 Phone List (带删除操作的静态trie树模板题)
题意:给出n个字符串,问是否满足所有的字符串都不以其他的字符串为前缀。
思路:字典树,先建树,然后每次查找的之前先删掉自己,找完以后再加回来。
以及这题动态建艹不过。。。学习了一下静态建树的写法。。。第一次写静态的写法。。。可以当做模板用。。。
1/* ***********************************************
2Author :111qqz
3Created Time :2016年08月16日 星期二 00时46分23秒
4File Name :code/poj/3630.cpp
5************************************************ */
6#include <cstdio>
7#include <cstring>
8#include <iostream>
9#include <algorithm>
10#include <vector>
11#include <queue>
12#include <stack>
13#include <set>
14#include <map>
15#include <string>
16#include <cmath>
17#include <cstdlib>
18#include <deque>
19#include <ctime>
20#define fst first
21#define sec second
22#define lson l,m,rt<<1
23#define rson m+1,r,rt<<1|1
24#define ms(a,x) memset(a,x,sizeof(a))
25typedef long long LL;
26#define pi pair < int ,int >
27#define MP make_pair
28using namespace std;
29const double eps = 1E-8;
30const int dx4[4]={1,0,0,-1};
31const int dy4[4]={0,-1,1,0};
32const int inf = 0x3f3f3f3f;
33const int N=1E5+7;
34int n;
35char s[N][12];
36int tot;
37struct Trie
38{
39 int nxt[10];
40 int cnt;
1 void init()
2 {
3 ms(nxt,0);
4 cnt = 0;
5 }
6}trie[N];
7int add()
8{
9 memset(&trie[tot],0,sizeof(Trie));
10 return tot++;
11}
12void Insert(char *s)
13{
14 int rt = 0 ;
15 int len = strlen(s);
16 for ( int i = 0 ; i < len ; i++)
17 {
18 int v = s[i]-'0';
19 if (!trie[rt].nxt[v]) trie[rt].nxt[v] = add();
20 rt = trie[rt].nxt[v];
21 trie[rt].cnt++;
22 }
23}
24void Delete(char *s)
25{
26 int rt = 0 ;
27 int len = strlen(s);
28 for ( int i = 0 ; i < len ; i++)
29 {
30 int v = s[i]-'0';
31 if (trie[rt].nxt[v])
32 {
33 rt = trie[rt].nxt[v];
34 trie[rt].cnt--;
35 }
36 }
37}
38bool Find(char *s)
39{
40 int rt = 0 ;
41 int len = strlen(s);
42 for ( int i = 0 ; i < len ;i++)
43 {
44 int v = s[i]-'0';
45 if (!trie[rt].nxt[v]||trie[trie[rt].nxt[v]].cnt==0) return false;
46 rt = trie[rt].nxt[v];
47 }
48 return true;
49}
50int main()
51{
52 #ifndef ONLINE_JUDGE
53 freopen("code/in.txt","r",stdin);
54 #endif
55 int T ;
56 cin>>T;
57 while (T--)
58 {
59 scanf("%d",&n);
60 tot = 1;
61 trie[0].init();
62 for ( int i = 1 ; i <= n ; i++)
63 {
64 scanf("%s",s[i]);
65 // cout<<"s[i]:"<<s[i]<<endl;
66 Insert(s[i]);
67 }
68 bool ok = true;
69 for ( int i = 1 ; i <= n ; i++)
70 {
71 Delete(s[i]);
72 if (Find(s[i]))
73 {
74 ok = false;
75 break;
76 }
77 Insert(s[i]);
78 }
79 if (ok)
80 {
81 puts("YES");
82 }
83 else
84 {
85 puts("NO");
86 }
87 }
88 #ifndef ONLINE_JUDGE
89 fclose(stdin);
90 #endif
91 return 0;
92}