hdu2846(2009多校第四场) 字典树
明明就是个简单的字典树，我不过就是想复习一下尼玛被自己坑死了T^T
把字符串分解那儿写错了，检查了半天没找到错误，然后以为被卡数组内存，难过的学习字典树的指针写法。。
发现还是不对，大半夜突然看到自己傻乎乎的错误顿时感觉好傻逼啊啊啊！！
字典树数组写法：
1 #include<stdio.h> 2 #include<string.h> 3 int flag[500005],count[500005],sz; 4 int ch[500005][30]; 5 char s[25]; 6 void insert(char *s,int v) 7 { 8 int u=0,n=strlen(s),i,c; 9 for (i=0;i<n;i++) 10 { 11 c=s[i]-'a'; 12 if (!ch[u][c]) 13 { 14 flag[++sz]=v; 15 count[sz]=1; 16 ch[u][c]=sz; 17 } 18 else if (flag[ch[u][c]]!=v) 19 { 20 count[ch[u][c]]++; 21 flag[ch[u][c]]=v; 22 } 23 u=ch[u][c]; 24 } 25 return; 26 } 27 int find(char *s) 28 { 29 int i,u=0,len=strlen(s),c; 30 for (i=0;i<len;i++) 31 { 32 c=s[i]-'a'; 33 if (!ch[u][c]) return 0; 34 u=ch[u][c]; 35 } 36 return count[u]; 37 } 38 int main() 39 { 40 int n,i,len,j; 41 while (~scanf("%d",&n)) 42 { 43 sz=0; 44 memset(ch,0,sizeof(ch)); 45 memset(count,0,sizeof(count)); 46 for (i=1;i<=n;i++) 47 { 48 scanf("%s",s); 49 len=strlen(s); 50 for (j=0;j<len;j++) 51 insert(s+j,i); 52 } 53 scanf("%d",&n); 54 for (i=1;i<=n;i++) 55 { 56 scanf("%s",s); 57 printf("%d\n",find(s)); 58 } 59 } 60 return 0; 61 }
指针写法：
1 #include<stdio.h> 2 #include<string.h> 3 struct tree{ 4 struct tree *son[26]; 5 int count,flag; 6 }*root; 7 void insert(char *s,int v) 8 { 9 int len=strlen(s),i,j,c; 10 tree *cur=root,*next; 11 for (i=0;i<len;i++) 12 { 13 c=s[i]-'a'; 14 if (cur->son[c]!=NULL) cur=cur->son[c]; 15 else 16 { 17 next=new(tree); 18 for (j=0;j<26;j++) 19 next->son[j]=NULL; 20 next->count=0; next->flag=-1; 21 cur->son[c]=next; 22 cur=next; 23 } 24 if (cur->flag!=v) 25 { 26 cur->count++; 27 cur->flag=v; 28 } 29 } 30 return; 31 } 32 int find(char *s) 33 { 34 int i,len=strlen(s),c; 35 tree *cur=root; 36 for (i=0;i<len;i++) 37 { 38 c=s[i]-'a'; 39 if (cur->son[c]==NULL) break; 40 cur=cur->son[c]; 41 } 42 if (i<len) return 0; 43 return cur->count; 44 } 45 int main() 46 { 47 int n,i,len,j,m; 48 char s[25],st[25]; 49 while (~scanf("%d",&n)) 50 { 51 root=new(tree); 52 for (i=0;i<26;i++) 53 root->son[i]=NULL; 54 root->count=0; root->flag=-1; 55 for (i=1;i<=n;i++) 56 { 57 scanf("%s",s); 58 len=strlen(s); 59 for (j=0;s[j];j++) 60 { 61 // memcpy(st,s+j,len-j); 62 insert(s+j,i); 63 } 64 } 65 scanf("%d",&m); 66 for (i=1;i<=m;i++) 67 { 68 scanf("%s",s); 69 printf("%d\n",find(s)); 70 } 71 } 72 return 0; 73 }
