【HDU】1251统计难题 （字典树：二维数组，结构体数组，链表，map）

使用二维数组或者结构体数组都可以，但是在计数的时候有一点点小区别 

一、结构体数组

#include 
#include 
#include 
#include 
#include 
typedef long long ll;
using namespace std;
const int maxn = 500000;
typedef struct 
{
	int cnt;
	int next[26];
}node;
int tot;
node T[maxn];

void insert(char *s)
{
	int u =0;
	for(int i=0;s[i];i++)
	{
		int x=s[i]-'a';
		if(T[u].next[x]==0)
		{
			T[u].next[x]=++tot;
		}
		u=T[u].next[x];	
		T[u].cnt++;
	}
}

int find(char *s)
{
	int u = 0;
	for(int i=0;s[i];i++)
	{
		int x = s[i]-'a';
		if(T[u].next[x]==0)
			return 0;
		u=T[u].next[x];
	}
	return T[u].cnt;
}


int main()
{
	char s[20];
	memset(T,0,sizeof(T));
	tot=0;
    while(gets(s))
    {
    	if(s[0]=='\0')
    		break;
    	insert(s);	
	}
	while(scanf("%s",s)!=EOF)
	{
		printf("%d\n",find(s));
	}
    return 0;
}

二、二维数组 

#include
#include
#include
#include
#define maxn 2000005
using namespace std;
int tot=0,n;
int trie[maxn][26];
int sum[maxn];
void insert(char *s)
{
	int i,root=0;
	for(i=0;s[i];i++)
	{
		int x=s[i]-'a';
		if(trie[root][x]==0)
		{
			trie[root][x]=++tot;
		}
		sum[trie[root][x]]++;
		root = trie[root][x];
	}
}

int find(char *s)
{
	int i,root=0;
	for(i=0;s[i];i++)
	{
		int x=s[i]-'a';
		if(trie[root][x]==0)
			return 0;
		root = trie[root][x];
	}
	return sum[root];
}
char s[20];
int main ()
{
	int i,root=0;
	while(gets(s))
	{
		if(s[0]=='\0')
			break;
		insert(s);
	}
	while(scanf("%s",s)!=EOF)
	{
		printf("%d\n",find(s)); 
	}
	return 0;
}

三、链表，指针（C++提交/G++显示超内存）

#include 
#include 
#include 
#include 
#include 
using namespace std;
const int maxn = 26;
 
typedef struct node
{
	int v;
	struct node *next[maxn];
	node()
	{
		v = 1;
		for(int i=0;i<26;i++)
			next[i]=NULL;
	}
}N;

N *root;

void insert(char *s)
{
	N *p = root;
	for(int i=0;s[i];i++)
	{
		int x = s[i]-'a';
		if(p->next[x]==NULL)
			p->next[x] = new N;
		else
			p->next[x]->v++;
		p=p->next[x];
	}
} 


int find(char *s)
{
	N *p=root;
	for(int i=0;s[i];i++)
	{
		int x = s[i]-'a';
		p=p->next[x];
		if(p==NULL)
			return 0;
	}
	return p->v;
}

 
int main ()
{
	char str[30];
	root = new N;
	while(gets(str))
	{
		if(str[0]=='\0')
			break;
		insert(str);
	}	
	while(gets(str))
		printf("%d\n",find(str));
	return 0;
}

四、map

#include 
#include 
#include 
#include 
#include 
#include 

 

（G++ AC / C++ WA） 

#include 
#include 
#include