統計難題

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


Problem DescriptionIgnatius最近遇到一個難題,老師交給他很多單詞(只有小寫字母組成,不會有重複的單詞出現),現在老師要他統計出以某個字串為字首的單詞數量(單詞本身也是自己的字首).

Input輸入資料的第一部分是一張單詞表,每行一個單詞,單詞的長度不超過10,它們代表的是老師交給Ignatius統計的單詞,一個空行代表單詞表的結束.第二部分是一連串的提問,每行一個提問,每個提問都是一個字串.

注意:本題只有一組測試資料,處理到檔案結束.

Output對於每個提問,給出以該字串為字首的單詞的數量.

Sample Inputbanana band bee absolute acm ba b band abc
Sample Output2 3 1 0
AuthorIgnatius.L
Recommend

Ignatius.L   |   We have carefully selected several similar problems for you:  

1075 1247 1671 1298 1800 

                一道字典樹基礎題把~~需要注意的部分寫在程式碼上了~

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
using namespace std;
int tree[500001][26], cnt, sum[500001];//之前開了20w結果re了~~
int n, m;
char s[100];
void insert()
{
	int len = strlen(s);
	int rt = 0;
	for (int i = 0; i<len; i++)
	{
		int id = s[i] - 'a';//這個節點的兒子~
		if (!tree[rt][id])//如果沒有出現過~
		{
			tree[rt][id] = ++cnt;//創造一個~
		}
		sum[tree[rt][id]]++;//某個匹配成功的字首數量是下個位置的sum~
		rt = tree[rt][id];//進位~進入下個節點
	}
}
int search()
{
	int rt = 0;
	int len = strlen(s);
	for (int i = 0; i<len; i++)
	{
		int id = s[i] - 'a';
		if (!tree[rt][id]) return 0;//此時匹配失敗~返0退出
		rt = tree[rt][id];//否則繼續進位查詢
	}//rt經過此迴圈後變成字首最後一個字母所在位置的後一個位置
	return sum[rt];//因為字首後移了一個儲存，所以此時的sum[rt]就是要求的前綴出現的次數
}
void init()
{
	memset(tree, 0, sizeof(tree));
	memset(sum, 0, sizeof(sum));
}
int main()
{
	init();
	while(1)
	{
		gets(s);
		if(s[0]==NULL)
        {
            break;
        }
        insert();
	}
	while(gets(s))
	{
		printf("%d\n", search());
	}
	return 0;
}