难题--杭电1251 解题报告

                                      统计难题

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 58046    Accepted Submission(s): 20262
 

Problem Description

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

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
注意:本题只有一组测试数据,处理到文件结束.

 

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2

3

1

0

此题为模板题。建一棵字典树即可。 博主也厚着脸皮盗来一张图：

 

上代码：

#include <iostream>
#include <string>
#include <cstring>
using namespace std;
struct node
{
	int cnt;
	node *next[26];
	node()
	{
		cnt=0;
		//memset(next,NULL,sizeof(next));
		for(int i=0;i<26;i++)
		next[i]=NULL;
	}
}root;
void CreatTree(string str)
{
	node *p=&root;
	int id,len=str.size();
	for(int i=0;i<len;i++)
	{
		id=str[i]-'a';
		if(p->next[id]==NULL)
		{
			p->next[id]=new node();
		}
		p=p->next[id];
		p->cnt++;
	}
}
int find(string str)
{
	node *p=&root;
	int id,len=str.size();
	for(int i=0;i<len;i++)
	{
		id=str[i]-'a';
		p=p->next[id];
		if(p==NULL)
		{
			return 0;
		}
	}
	return p->cnt;
}
int main()
{
	string s;
	do
	{
		getline(cin,s);
		CreatTree(s);
	}while(s=="");
	while(cin>>s)
	{
		cout<<find(s)<<endl;
	}
}

 当然还可以用string 加map 做

上代码：

#include <iostream>
#include <string>
#include <map>
using namespace std;
int main()
{
     map<string,int> m;
     string str;
     do
     {
     	getline(cin,str);
     	int len=str.size();
     	for(int i=1;i<=len;i++)
     	{
     		m[str.substr(0,i)]++;
		}
	 }while(str!="");
	 while(cin>>str)
	 cout<<m[str]<<endl;
}

第二种方法代码短。而且map内部用的是红黑二叉树所以速度更快。

然而知道字典树的底层写法也是很重要的。(๑╹◡╹)ﾉ"""