Trie字符串统计

const int N=10010;

int cnt[N];//查找单词数量
int son[N][26];//子节点
int idx;
//下标是0的点，既是根节点，又是空节点 

void insert(char str[]) 
{
	int p=0;
	for(int i=0;str[i];i++)
	{
		int u=str[i]-'a';
		if(!son[p][u]) son[p][u]=++idx;
		p=son[p][u];
	}
	cnt[p]++;//遇到单词最后一个字母，单词数量++ 
}

int query(char str[])
{
	int p=0;
	for(int i=0;str[i];i++)
	{
		int u=str[i]-'a';
		if(!son[p][u]) return 0;
		p=son[p][u];
	}
	return cnt[p];//返回单词数量 
}