POJ 3630

View Code

 1 /*
 2 trie树：
 3 利用trie树 查找单词 的简单程序 
 4 */ 
 5 //法一：自己的方法 
 6 #include<iostream>
 7 #include<cstdio>
 8 using namespace std;
 9 
10 struct trie
11 {
12    bool is;//标记是否是个单词 
13    trie *next[11];// 分别代表
14    void (*init)(trie *);// 为了初始化is=0，next=NULL 
15 }node[100000];
16 
17 trie *root;
18 bool f;
19 int k;
20 
21 void teinit(trie *y)
22 {
23   int i;
24   (*y).is=0;
25   for(i=0;i<11;++i)
26    (*y).next[i]=NULL; 
27 }
28 
29 void insert(const char *word)
30 {
31   int i;
32   trie *p=root;
33   while(*word)
34   {
35     if(p->next[*word-'0']==NULL)
36      {
37         p->next[*word-'0']=&node[k++];
38      } 
39      p=p->next[*word-'0'];
40      if(p->is)f=1;// 检测 别的号码 是这个号码的前缀 如：输入 1 12 型数据 
41      word++;
42   }
43   for(i=0;i<11;++i)//  检测此 号码是别的号码的前缀   如：输入  12 1 型数据 
44    if(p->next[i]!=NULL)f=1;  
45   p->is=true;//单词遍历结束 标记一下说明到此处有个单词 
46 }
47 
48 
49 int main()
50 {
51    char ch[80];
52    int i,n,t;
53    scanf("%d",&t);
54    while(t--)
55    {
56      for(i=0;i<100000;++i)
57      {
58        node[i].init=teinit;
59        node[i].init(&node[i]);
60       }
61      root=&node[0]; 
62      k=1;
63      f=0;
64      scanf("%d",&n);
65      for(i=0;i<n;++i)
66       {
67         scanf("%s",ch);
68         insert(ch);
69       }
70      if(f)printf("NO\n");
71      else printf("YES\n");
72    }
73    system("pause");
74    return 0;
75 }

法二：排序  +memset()

View Code

 1 /*
 2 trie树：
 3 利用trie树 查找单词 的简单程序 
 4 */ 
 5 //法一：自己的方法 
 6 #include<iostream>
 7 #include<cstdio>
 8 #include<algorithm>
 9 using namespace std;
10 
11 struct trie
12 {
13    bool is;//标记是否是个单词 
14    trie *next[10];// 分别代表
15 }node[100000];
16 
17 struct node
18 {
19   int l;
20   char ch[11];
21 }str[10001];
22 
23 trie *root;
24 bool f;
25 int k;
26 
27 bool cmp(struct node x,struct node y)
28 {
29    return x.l<y.l;
30 }
31 
32 void insert(const char *word)
33 {
34   int i;
35   trie *p=root;
36   while(*word)
37   {
38     if(p->next[*word-'0']==NULL)
39      {
40         p->next[*word-'0']=&node[k++];
41         memset(p->next[*word-'0'],0,sizeof(trie));
42      }
43      p=p->next[*word-'0'];
44      if(p->is)f=1;
45      word++;
46   }
47   p->is=true;
48 }
49 
50 
51 int main()
52 {
53    int i,n,t,len;
54    scanf("%d",&t);
55    while(t--)
56    {
57      root=&node[0]; 
58      memset(root,0,sizeof(trie));
59      k=1;
60      f=0;
61      scanf("%d",&n);
62      for(i=0;i<n;++i)
63      {
64       scanf("%s",str[i].ch);
65       len=strlen(str[i].ch);
66       str[i].l=len;
67      }
68      sort(str,str+n,cmp); 
69      for(i=0;i<n;++i)
70       {
71         insert(str[i].ch);
72       }
73      if(f)printf("NO\n");
74      else printf("YES\n");
75    }
76    system("pause");
77    return 0;
78 }