HDOJ 1251 统计难题——第二次用字典树AC题目，写一下解题报告

    第一次用字典树+BFS的方法统计以给定字符串为前缀的单词数目，超时了。百思不得其解，然后我看了一下讨论版里的一位同学的AC代码。豁然开朗，立刻明白了求解相同前缀的单词个数的最简单的解法。

    其实在建一科字典树的时候，每插入一个字母，就相当于给以当前已经插入的串为前缀的单词增加了一个。这样，统计的时候就不用BFS了。哈哈。废话不多说了，贴出我的代码，跟大家分享，造福后代。

#include<iostream>
#include<string.h>
#include<queue>
using namespace std;

const int Branch = 26;

struct Trie
{
	int count;
	Trie * next[Branch];
};	//字典树结点


//
void insert(Trie * root, char * str)
{
	Trie * p = root;
	while(*str)
	{
		int seqNum = *str - 'a';
		if(!p->next[seqNum])
		{
			p->next[seqNum] = new Trie();
			memset(p->next[seqNum], 0, sizeof(Trie));
		}
		p = p->next[seqNum];
		p->count ++;

		str++;	//忘记写这句话，导致内存耗尽，最终死机
	}
	//p->tail = true;
}

//
int statistics(Trie * root, char * str)
{
	Trie * p = root;

	while(*str)
	{
		int seqNum = *str - 'a';
		if(p->next[seqNum] == NULL) return 0;
		else 
		{
			p = p->next[seqNum];
			str ++;
		}
	}

	int sum = p->count;
	/*
	int sum = 0;
	queue<Trie *> q;
	q.push(p);
	while(!q.empty())
	{
		Trie * cur = q.front();	q.pop();
		if(cur->tail) sum ++;
		for(int i=0; i<Branch; i++)
			if(cur->next[i]) q.push(cur->next[i]);
	}*/
	return sum;
}


int main()
{
	char word[11];
	Trie * root = new Trie();
	memset(root, 0, sizeof(Trie));

	while(gets(word))
	{
		if(word[0] == '\0') break;

		insert(root, word);
	}

	while(scanf("%s", word) != EOF)
	{
		int sum = statistics(root, word);
		cout << sum << endl;
	}
	system("pause");
	return 0;
}

       代码里那段被注释掉的代码块是之前BFS求解相同前缀数目的解法。也是一种求解方法，保留了。哈哈。第二次AC字典树的问题。