hdu2846 Repository 字典树（好题）

把每个字符串的所有子串都加入字典树，但在加入时应该注意同一个字符串的相同子串只加一次，因此可以给字典树的每个节点做个记号flag--表示最后这个前缀是属于那个字符串，如果当前加入的串与它相同，且二者属于不同串就自加，否则不作处理。

AC代码：

#include<cstdio>
#include<cstring>
const int Max=26;
struct node{
	node *next[Max];
	int cnt,num; //用num记录最后一次加入的字符串编号,若不同串则自加
	node(){
		for(int i=0;i<26;++i) next[i]=NULL;
		cnt=num=0;
	}
}*root;
void insert_tree(int num,char *s,int n){
	node *p=root,*q;
	for(int i=0;i<n;++i){
		int u=s[i]-'a';
		if(p->next[u]==NULL){
			q=new node();
			q->cnt=1;
			q->num=num;
			p->next[u]=q;
			p=p->next[u];
		}
		else{
			if(num!=p->next[u]->num) {
				p->next[u]->cnt++;
				p->next[u]->num=num;
			}
			p=p->next[u];
		}
	}
}
int getAns(char *s){
	int n=strlen(s);
	node *p=root;
	for(int i=0;i<n;++i){
		int u=s[i]-'a';
		if(p->next[u]==NULL) return 0;
		p=p->next[u];
	}
	return p->cnt;
}
void free_tree(node *u){
	if(u==NULL) return;
	for(int i=0;i<26;++i) free_tree(u->next[i]);
	delete u;
}
int main(){
	int n,m;
	char s[40],s1[40];
	scanf("%d",&n);
	root=new node();
	for(int i=0;i<n;++i){
		scanf("%s",s);
		int len=strlen(s);
		for(int j=0;j<len;++j){
			int c=0;
			for(int k=j;k<len;++k) s1[c++]=s[k];
			s1[c]='\0';
			insert_tree(i,s1,c);
		}
	}
	scanf("%d",&m);
	for(int i=0;i<m;++i) {
		scanf("%s",s);
		printf("%d\n",getAns(s));
	}
	free_tree(root);
	return 0;
}

如有不当之处欢迎指出！