HDU 1251-统计难题-基本字典树

最新推荐文章于 2020-07-23 01:46:41 发布

原创最新推荐文章于 2020-07-23 01:46:41 发布 · 457 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构字典树专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种使用字典树结构高效查询包含特定前缀的单词数量的方法，通过遍历字典树实现快速查找。

给n个单词建立一个表

给n个串

查询以该串为前缀的单词个数包括本身

就是单纯地遍历一次字典树

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map> 
#include <vector>
using namespace std;
#define inf 2147483647  
const int maxnode = 400110;
  
 
	int len; 
struct trie
{
	int ch[maxnode][26];
	int val[maxnode];
	int sz;
	void init()
	{
		sz=1;
		val[0]=0;		//可不需要
		memset(ch[0],0,sizeof(ch[0]));
	}
	int idx(char c) {return c-'a';}
	
	void insert(char *s )
	{
		
		int u=0,n=strlen(s),i;
		for (i=0;i<n;i++)
		{
			int c=idx(s[i]);
			if (!ch[u][c])
			{
				memset(ch[sz],0,sizeof(ch[sz]));
				val[sz]=0;
				ch[u][c]=sz++;
			}
			u=ch[u][c];
			val[u]++;
		} 
	}
 
	int query_if(char *s)	//查找当前单词是否存在,本题用不上
	{
		int u=0,n=strlen(s),i;
		for (i=0;i<n;i++)
		{
			int c=idx(s[i]);
			if (!ch[u][c])
				return 0;
			else
			{
				u=ch[u][c];
			} 
		}
		return val[u]; 
	}
  
};
trie tp;
char tm[15];
int main()
{ 
		tp.init();  
	while( gets(tm))
	{
		if (tm[0]==0)
			break;		
			tp.insert(tm);  
		
	}
	
	while( scanf("%s", &tm)!=EOF)
	{ 
		printf("%d\n",tp.query_if(tm )); 
	  
	} 

	return 0;
}