hdu1251——统计难题

最新推荐文章于 2019-03-18 20:48:45 发布

原创最新推荐文章于 2019-03-18 20:48:45 发布 · 421 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#前缀树

前缀树专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了裸前缀树的概念、原理及其实现过程，通过代码实例展示了如何使用裸前缀树解决字符串匹配问题。文章还强调了在理解和应用裸前缀树时的注意事项，并提供了代码优化建议。

裸前缀树只不过指针和动态内存我比较陌生以前没用过啊QAQ

熟悉语法水一日志告诉自己就算noip洗白了也不要放弃

#include<cstdio>
#include<cstring>
#include<malloc.h>

#define MAXN (30)
#define cle(a,b) memset(a,b,sizeof(a))
#define loop(i,j,k,l) for(int i=j;l>0?i<=k:i>=k;i+=l)

struct trie {
	trie *next[MAXN];
	int v;
}*root;

char s[MAXN];int len;

int id;

void create() {
	trie *p=root,*q;
	
	loop(i,0,len,1) {
		id=s[i]-'a'+1;
		if(p->next[id]==NULL) {
			q=(trie*)malloc(sizeof(trie));
			loop(j,1,26,1) {
				q->next[j]=NULL;
			}
			q->v=1;	
			p->next[id]=q;
				
		}
		else{
			p->next[id]->v++;
		}
		p=p->next[id];			
		
			
	}
	
}

void readdata() {	
	gets(s);
	len=strlen(s)-1;
	while(s[0]!=0) {
		create();
		cle(s,0);
		gets(s);
		len=strlen(s)-1;
	}
}

int find() {
	trie *p=root;
	
	loop(i,0,len,1) {
		id=s[i]-'a'+1;
		if(p->next[id]==NULL){
			return 0;
		}
		p=p->next[id];
		if(i==len) {
			return p->v;
		}
	}
}

void solve() {
	while(scanf("%s",s)!=EOF) {
		len=strlen(s)-1;
		
		printf("%d\n",find());
		
		
		cle(s,0);
	}
}
int main() {
	root=(trie*)malloc(sizeof(trie));
	loop(i,1,26,1) {
		root->next[i]=NULL;
	}
	readdata();
	solve();
}