专题训练之字典树
推荐几篇博客：https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html 浅谈Trie树
https://blog.csdn.net/u013588639/article/details/38406453 字典树的数组实现
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int maxn=200010; 6 int ch[maxn][27],val[maxn]; 7 int sz; 8 9 void init() 10 { 11 sz=0; 12 memset(ch,0,sizeof(ch)); 13 memset(val,0,sizeof(val)); 14 } 15 16 void insert(char *s) 17 { 18 int u=0,c; 19 for ( int i=0;s[i];i++ ) 20 { 21 c=s[i]-'a'; 22 if ( !ch[u][c] ) ch[u][c]=++sz; 23 u=ch[u][c]; 24 val[u]++; 25 } 26 } 27 28 int query(char *s) 29 { 30 int u=0,c; 31 for ( int i=0;s[i];i++ ) 32 { 33 c=s[i]-'a'; 34 if ( !ch[u][c] ) return 0; 35 u=ch[u][c]; 36 } 37 return val[u]; 38 }
练习题：
1.（HDOJ1251）http://acm.hdu.edu.cn/showproblem.php?pid=1251
分析：模板题，注意数组要开大不然会RE
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int maxn=500010; 6 int ch[maxn][27],val[maxn]; 7 int sz; 8 char s[15]; 9 10 void init() 11 { 12 sz=0; 13 memset(ch,0,sizeof(ch)); 14 memset(val,0,sizeof(val)); 15 } 16 17 void insert(char *s) 18 { 19 int u=0,c; 20 for ( int i=0;s[i];i++ ) 21 { 22 c=s[i]-'a'; 23 if ( !ch[u][c] ) ch[u][c]=++sz; 24 u=ch[u][c]; 25 val[u]++; 26 } 27 } 28 29 int query(char *s) 30 { 31 int u=0,c; 32 for ( int i=0;s[i];i++ ) 33 { 34 c=s[i]-'a'; 35 if ( !ch[u][c] ) return 0; 36 u=ch[u][c]; 37 } 38 return val[u]; 39 } 40 41 int main() 42 { 43 int i,j,k; 44 bool flag=true; 45 init(); 46 while ( gets(s) ) 47 { 48 if ( flag ) 49 { 50 if ( strlen(s)!=0 ) insert(s); 51 else flag=false; 52 } 53 else printf("%d\n",query(s)); 54 } 55 return 0; 56 }
2.（HDOJ1671）http://acm.hdu.edu.cn/showproblem.php?pid=1671
题意：是否有一个电话号码是另一个电话号码的前缀
分析：可以将更新和查询放到同一个数组中进行，开一个bool型的数组isw记录该编号的节点是否为单词的结尾（可以解决前缀先输入的情况）。而对于前缀后输入的情况，在插入时设置一个bool型的变量ok，判断在整个插入过程是否有申请过新的节点
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int maxn=5e5+10; 6 int ch[maxn][11]; 7 int sz; 8 bool isw[maxn],flag; 9 char s[105]; 10 11 void init() 12 { 13 sz=0; 14 flag=false; 15 memset(ch,0,sizeof(ch)); 16 memset(isw,false,sizeof(isw)); 17 } 18 19 void insert(char *s) 20 { 21 int u=0,c; 22 bool ok=false; 23 for ( int i=0;s[i];i++ ) 24 { 25 c=s[i]-'0'; 26 if ( !ch[u][c] ) { 27 ch[u][c]=++sz; 28 ok=true; 29 } 30 u=ch[u][c]; 31 if ( isw[u] ) flag=true; 32 } 33 isw[u]=true; 34 if ( !ok ) flag=true; 35 } 36 37 int main() 38 { 39 int T,n,m,i,j,k,x,y,z; 40 scanf("%d",&T); 41 while ( T-- ) 42 { 43 scanf("%d",&n); 44 init(); 45 for ( i=1;i<=n;i++ ) 46 { 47 scanf("%s",s); 48 insert(s); 49 } 50 if ( flag ) printf("NO\n"); 51 else printf("YES\n"); 52 } 53 return 0; 54 }
3.（HDOJ1247）http://acm.hdu.edu.cn/showproblem.php?pid=1247
题意：给定若干个字符串，找出这样的字符串——它可由其他两个字符串拼接得到。
法1：
分析：先将所有单词存入字典树中，然后对每个单词枚举可以分成的两部分形式（用cstring中的函数strncpy）进行判断这两部分是否都存在于树中
介绍：strncpy()：strncpy(dest,src,n); strncpy把src所指向以'\0'结尾的字符串的前n个字符复制到dest所指的数组中，返回指向dest的指针。
当n>=sizeof(src)时，拷贝正确，并在dest字符串后面加入'\0';
当n<sizeof(src)时，只拷贝src前n-1个字符串到dest，不会为dest字符串后面加入'\0';
注意：可能存在hat hathat这样的情况即一个单词用两次拼成另一个单词；单词长度未知，数组尽量开大;对于存两部分的数组初始化为"\0"这样的形式
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 using namespace std; 5 const int maxn=200010; 6 int ch[maxn][27]; 7 bool val[maxn]; 8 int sz; 9 char s[maxn][105]; 10 11 12 void init() 13 { 14 sz=0; 15 memset(ch,0,sizeof(ch)); 16 memset(val,false,sizeof(val)); 17 } 18 19 void insert(char *s) 20 { 21 int u=0,c; 22 for ( int i=0;s[i];i++ ) 23 { 24 c=s[i]-'a'; 25 if ( !ch[u][c] ) ch[u][c]=++sz; 26 u=ch[u][c]; 27 } 28 val[u]=true; 29 } 30 31 bool query(char *s) 32 { 33 int u=0,c; 34 for ( int i=0;s[i];i++ ) 35 { 36 c=s[i]-'a'; 37 if ( !ch[u][c] ) return false; 38 u=ch[u][c]; 39 } 40 return val[u]; 41 } 42 43 int main() 44 { 45 int cnt=0,n,i,j,k,m,x,y,z,ans,len; 46 init(); 47 while ( scanf("%s",s[++cnt])!=EOF ) insert(s[cnt]); 48 for ( i=1;i<=cnt;i++ ) 49 { 50 len=strlen(s[i]); 51 char str1[105]="\0"; 52 char str2[105]="\0"; 53 for ( j=1;j<len-1;j++ ) 54 { 55 strncpy(str1,s[i],j); 56 strncpy(str2,s[i]+j,len-j+1); 57 if ( query(str1) && query(str2) ) 58 { 59 printf("%s\n",s[i]); 60 break; 61 } 62 } 63 } 64 return 0; 65 }
不用strnpy分解一个单词（改动模板）
1 #include<cstdio> 2 #include<cstring> 3 using namespace std; 4 int ch[200010][27]; 5 int sz; 6 int val[200010]; 7 char str[50005][20]; 8 void init() 9 { 10 sz=0; 11 memset(ch,0,sizeof(ch)); 12 memset(val,0,sizeof(val)); 13 } 14 15 void insert(char* s) 16 { 17 int u=0,c,i; 18 for ( i=0;i<strlen(s);i++ ) { 19 c=s[i]-'a'; 20 if ( !ch[u][c] ) ch[u][c]=++sz; 21 u=ch[u][c]; 22 } 23 val[u]=1; 24 } 25 26 bool query(char* s) 27 { 28 int u=0,c,i,j,U,C; 29 for ( i=0;i<strlen(s);i++ ) { 30 c=s[i]-'a'; 31 if ( !ch[u][c] ) return false; 32 u=ch[u][c]; 33 if ( val[u]==1 && i+1<strlen(s) ) { 34 U=0; 35 for ( j=i+1;j<strlen(s);j++ ) { 36 C=s[j]-'a'; 37 if ( !ch[U][C] ) break; 38 U=ch[U][C]; 39 } 40 if ( val[U]==1 && j==strlen(s) ) return true; 41 42 } 43 } 44 return false; 45 } 46 47 int main() 48 { 49 int m,n,i,j,k,cnt=0; 50 init(); 51 while ( scanf("%s",str[cnt])!=EOF ) { 52 insert(str[cnt]); 53 cnt++; 54 } 55 for ( i=0;i<cnt;i++ ) { 56 if ( query(str[i]) ) printf("%s\n",str[i]); 57 } 58 return 0; 59 }
