POJ 3630 Phone List 字典树

         今晚学了字典树，找来一题熟悉熟悉，直接套模板，一开始在Sicily过了，但在POJ 超时，囧，看了别人的解题报告，直接开个10000数组来储存节点，而不是插入时再new一个，就这样AC掉了

#include <iostream>
#include <stdio.h>
#include <memory.h>
#include <string.h>

using namespace std;

int nodenum;
bool find_prex;

struct Trie_node
{
	bool isStr;
	Trie_node *next[10];
	Trie_node()
	{
		isStr = false;
		memset(next, NULL, sizeof(next));
	}
};

Trie_node node[100000];

class Trie
{
public:
	Trie();
	void clear();
	void insert(char *);
private:
	Trie_node *root;
};

void Trie::clear()
{
	for (int i = 0; i < nodenum; i++)
	{
		node[i].isStr = false;
		memset(node[i].next, NULL, sizeof(node[i].next));
	}
}

Trie::Trie()
{
	root = &node[0];
}

void Trie::insert(char * word)
{
	//当找到一个number不符合题意的话，就不做插入操作
	if (find_prex) return;  
	
	Trie_node *p = root;

	while (*word)
	{
		//当在树中找到一个单词是插入单词的前缀时，停止查找
		if (p->isStr)
		{	
			find_prex = true; 
			return ;
		}

		if (p->next[*word-'0'] == NULL)
		{
			Trie_node *tmp = &node[nodenum];
			p->next[*word-'0'] = tmp;
			nodenum++;
		}
		p = p->next[*word-'0'];
		word++;
	}
	//判断树中是否有元素与插入的number相同
	if (p->isStr)
		find_prex = true;

	//判断树中是否有元素的前缀是插入单词
	for (int i = 0; i < 10; i++)
		if (p->next[i] != NULL)
		{
			find_prex = true;
			return ;
		}

	p->isStr = true;
}


int main()
{
	Trie t;
	int cases, n;
	char number[11];

	cin >> cases;

	while (cases--)
	{
		nodenum = 1;
		find_prex = false;
		cin >> n;

		for (int i = 0; i < n; i++)
		{
			scanf("%s", number);
			if (!find_prex)
				t.insert(number);
		}

		if (!find_prex)
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
		t.clear();
	}
}