字典树专题

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 #define maxn 500050
 4 char str[50010][100];
 5 int nxt;
 6 struct node
 7 {
 8     int val;
 9     int next[26];
10 }tree[maxn];
11 //将结构体改成数组更容易初始化，treenext[maxn][26],treeval[maxn] 
12 int add()
13 {
14     memset(&tree[nxt],0,sizeof(node));
15     return nxt++;
16 }
17 void insert(char *s)
18 {
19     int rt=0,len=strlen(s);
20     for(int i=0;i<len;i++)
21     {
22         int c=s[i]-‘a‘;
23         if(!tree[rt].next[c])
24         {
25             tree[rt].next[c]=add();
26         }
27         rt=tree[rt].next[c];
28     }
29     tree[rt].val++;
30 }
31 int query(char *s){
32     int rt=0, len = strlen(s);
33     int id=-1;
34     for(int i=0; i<len; i++){
35         int c=s[i]-‘0‘;
36         if(!tree[rt].next[c]) return -1;
37         rt=tree[rt].next[c];
38         id = tree[rt].val;
39     }
40     return id;
41 }
42 int main()
43 {
44     int l=0;
45     nxt=1;
46     while(scanf("%s",str[l])!=EOF)
47     {
48         insert(str[l]);
49         l++;
50     }
51     for(int i=0;i<l;i++)
52     {
53         printf("%d\n",query(str[i]));
54     }
55     return 0;
56 }

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 
 4 char c[100][100];
 5 const int MAX=10;
 6 typedef struct Node
 7 {
 8     int val;
 9     struct Node *next[MAX];
10 }TrieNode;
11 
12 TrieNode *head;
13 
14 void insert(char str[], int id)
15 {
16     Node *t,*s=head;
17     int len=strlen(str);
18     for(int i=0;i<len;i++)
19     {
20         int c=str[i]-‘0‘;
21         if(s->next[c]==NULL)
22         {
23             t=new Node;
24             for(int j=0;j<10;j++) t->next[j]=NULL;
25             t->val=-1;
26             s->next[c]=t;
27         }
28         s=s->next[c];
29         if(s->val<0) s->val=id;
30     }
31 }
32 
33 int query(char str[])
34 {
35     Node *s=head;
36     int ans;
37     int len=strlen(str);
38     for(int i=0;i<len;i++)
39     {
40         int c=str[i]-‘0‘;
41         if(s->next[c]==NULL) return -1;
42         s=s->next[c];
43         ans=s->val;
44     }
45     return ans;
46 }
47 
48 void Tree_Del(Node *p)
49 {
50     for(int i=0;i<10;i++)
51     {
52         if(p->next[i]!=NULL)
53           Tree_Del(p->next[i]);
54     }
55     free(p);
56 }
57 
58 int main()
59 {
60     head=new Node;
61     for(int i=0;i<10;i++)head->next[i]=NULL;
62     head->val=-1;
63     int n, m;
64     char str[60];
65     scanf("%d%d",&n, &m);
66     for(int i=0; i<n; i++)
67     {
68         scanf("%s",str);
69         insert(str, i);
70     }
71     for(int i=0; i<m; i++){
72         scanf("%s", str);
73         printf("%d\n", query(str));//输出字符串是在第几次出现过 
74     }
75     Tree_Del(head);//释放空间，防止超内存 
76     return 0;
77 }