hdu1251 字典树

题干:
Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

字典树模板,初始时是一棵空树,然后根据题干给出的数据开始建树,没节点就加节点。一层创建完后记得移动root节点,好进行下一层创建。


#include <bits/stdc++.h>
using namespace std;
int trie[400010][26],len,root,tot,sum[400010];
bool p;
char s[26];
void insert()
{
	len=strlen(s);
	root=0; 
	int i,j;
	for (i=0;i<len;i++)
	{
		int id=s[i]-'a';
		if (!trie[root][id])
		trie[root][id]=++tot;//无此节点,添加新节点 
		sum[trie[root][id]]++;//计数
		root=trie[root][id];//根变成这一层,方便下一层创建节点	 
	}
} 
int search()
{
	root=0;
	int i,j;
	len=strlen(s); 
	for (i=0;i<len;i++)
	{
		int id=s[i]-'a';
		if (!trie[root][id])
		return 0;//无此节点,返回0
		root=trie[root][id]; 
		//root经过此循环后变成前缀最后一个字母所在位置的后一个位置 
	}
	return sum[root]; 
}
int main()
{
	int i,j,f=0;
	while(gets(s))
	{
		if (strlen(s)==0)
		{
			f=1;
			continue;
		}
		if (!f)
		insert();
		else
		cout<<search()<<endl;
	}

	return 0;
}

posted @ 2021-02-21 21:27  索饮  阅读(38)  评论(0编辑  收藏  举报