(HDU) 1247 Hat's Words (字典树)

传送门

题目大意：给你字符串，输出其中能由其他两个字符串组合而成的字符串。

解题思路：建立两颗字典树，一颗正序，一颗反序，开一个vis数组，find的时候正序find一下，如果字符串的第i位有其他字符串flag标记，就vis[i]++,反序find的时候，同样不过是vis[len-i-2]++，标记正序的位置-1，最后从0~len-1，搜一遍，看有没有位置vis[i]>=2,有的话就说明该字符串可以被其他两个字符串组合而成。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int maxn=2e6+5;
int tree1[maxn][30],tree2[maxn][30];
//一正序，一反序 
bool flag1[maxn],flag2[maxn];
int vis[maxn];
int tot1=0,tot2=0;
void add1(char *s){
	int root=0,id,len=strlen(s);
	for(int i=0;i<len;++i){
		id=s[i]-'a';
		if(!tree1[root][id])	tree1[root][id]=++tot1;
		root=tree1[root][id];
	}
	flag1[root]=true;
} 
void add2(char *s){
	int root=0,id,len=strlen(s);
	for(int i=0;i<len;++i){
		id=s[i]-'a';
		if(!tree2[root][id])	tree2[root][id]=++tot2;
		root=tree2[root][id];
	}
	flag2[root]=true;
} 
void find1(char *s){
	int root=0,id,len=strlen(s);
	for(int i=0;i<len;++i){
		id=s[i]-'a';
		root=tree1[root][id];
		if(flag1[root])	vis[i]++;
	}
}
void find2(char *s){
	int root=0,id,len=strlen(s);
	for(int i=0;i<len;++i){
		id=s[i]-'a';
		root=tree2[root][id];
		if(flag2[root])	vis[len-i-2]++;
	}
}
char a[50005][100];
set<string> ans; 
int main(){
	int cnt=0;
	while(scanf("%s",a[cnt++])!=EOF){
		add1(a[cnt-1]);
		strrev(a[cnt-1]);
		add2(a[cnt-1]);
		strrev(a[cnt-1]);
	}
	
	for(int i=0;i<cnt;++i){
		int len=strlen(a[i]);
		for(int j=0;j<len;++j)	vis[j]=0;
		find1(a[i]);
		strrev(a[i]);
		find2(a[i]);			
		strrev(a[i]);
		for(int j=0;j<len;++j){
			if(vis[j]>=2){
				string tp=a[i];
				ans.insert(tp);
				break;
			}
		} 
	}
	for(auto it=ans.begin();it!=ans.end();++it){
		cout<<*it<<endl;
	}
	return 0;
}