Trie树入门 字典树模板,HDU1251模板题（注意输入）

今天抽时间学了一下字典树，比想象中的简单，也很好理解，先把模板贴上

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=2e5+5;
int cnt[maxn],trie[maxn][26],idx;
void insert(char s[])
{
	int u=0;
	for(int i=0;s[i];i++)
	{
		int v=s[i]-'a';
		if(!trie[u][v])
			trie[u][v]=++idx;
		u=trie[u][v];
	}
	cnt[u]++;
}
int query(char s[])
{
	int u=0;
	for(int i=0;s[i];i++)
	{
		int v=s[i]-'a';
		if(!trie[u][v])
			return 0;
		u=trie[u][v];
	}
	return cnt[u];
}
char s[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
		getchar();
    	char op;
    	scanf("%c%s",&op,s);
    	if(op=='I')
    		insert(s);
    	else printf("%d\n",query(s));
	}
    return 0;
}

然后就是 HDU1251这道题，是个 模板题，照着模板稍微改一下就好了，需要注意的是输入的地方，用scanf %s我是真的搞不了，不知道为啥 回车根本不管用，用gets可以，但是怕某些oj不让用

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=5e5+5;
int cnt[maxn],trie[maxn][26],idx;
void insert(char s[])
{
	int u=0;
	for(int i=0;s[i];i++)
	{
		int v=s[i]-'a';
		if(!trie[u][v])
			trie[u][v]=++idx;
		u=trie[u][v];
		cnt[u]++;
	}
}
int query(char s[])
{
	int u=0;
	for(int i=0;s[i];i++)
	{
		int v=s[i]-'a';
		if(!trie[u][v])
			return 0;
		u=trie[u][v];
	}
	return cnt[u];
}
char s[30];
int main()
{
    while(gets(s))
    {
    	if(s[0]==' '||strlen(s)==0)
    		break;
    	insert(s);
	}
	while(~scanf("%s",s))
	{
		printf("%d\n",query(s));
	}
    return 0;
}