hdu 1247 Trie入门题

点击打开链接

判断有多少单词恰好是由两个单词拼接成

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
using namespace std;
const int M=26;
char s[51000][50];
typedef struct Trie{
	Trie *next[M];//节点的子树
	bool word;//单词节点标记 
}Trie;
Trie root;
void insert(char *s)
{
	int len=strlen(s);
	Trie *p=&root,*q;
	for(int i=0;i<len;i++)
	{
		int id=s[i]-'a';//子树编号
		if(p->next[id]==NULL)
		{
			q=new Trie();
			for(int j=0;j<M;j++)
			{
				q->next[j]=NULL;	
			}
			p->next[id]=q;	
		}
	
		p=p->next[id];//移动到子节点	
	}
	p->word=true;//单词节点标记 
}
bool find2(int i,char *s)
{
	int len=strlen(s);
	Trie *p=&root;
	while(i<len)//后半段也为一个单词 
	{
		int son=s[i]-'a';
		if(p->next[son]==NULL)
		{
			return false;	
		}
		else
		{
			p=p->next[son];		
			if(p->word)
			{
				if(i+1==len)//该单词恰好由两个单词组成 
				{
					return true;
				}
			}
			i++;
		}
	}
	return false;
}
bool find(char *s)//查询复杂度为单词的长度  
{
	Trie *p=&root;
	int i=0,cur;
	int len=strlen(s);
	while(i<len)
	{
		int son=s[i]-'a';
		if(p->next[son]==NULL)
		{
			return false;	
		}
		else
		{
			p=p->next[son];
			if(p->word&&find2(i+1,s))//前半段为一个单词,此时判断后半段是否成立 
			{
				return true;
			}
			i++;
	
		}
	}
	return false;


}
int main()
{
	for(int i=0;i<M;i++)
	{
		root.next[i]=NULL;
		root.word=0;
	}
	int i=0;
	while(scanf("%s",s[i])!=EOF)
	{
		insert(s[i]);
		i++;
	}
	for(int k=0;k<i;k++)//枚举判断该单词是否有两个单词拼接成 
	{				//判断匹配前半段之后在匹配后半段即可 
		if(find(s[k]))
		{
			printf("%s\n",s[k]);
		}	
	}
	return 0;
}