poj 3630 Phone List (带删除操作的静态trie树模板题)

poj 3630 题目链接

题意:给出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}