upc 单词表 字典树 + dfs

单词表
时间限制: 1 Sec 内存限制: 128 MB

题目描述
hzwer获得了一个n个单词的单词表，其中每个字符都是小写字母，现在，他想和他的妹子研究一下这个单词表。
设编号为u的单词与编号为v的单词（u <v ）构成单词对(u ,v)，记两个单词最长公共前缀为S1(u ,v)，最长公共后缀为S2(u ,v)。
hzwer从S1入手，他想知道，所有单词对生成的S1串中，长度最大的串，如果有多个，那么他在其中选择字典序最小的，并且你要告诉他S1串为该串的单词对数，而他的妹子则关心S2，要求类似。
但是hzwer要和他的妹子去度假，所以这个问题就交给你了。
保证至少存在一个长度大于0的S1串和S2串。
输入
第一行一个正整数n，表示单词个数。
接下来n行，每行一个字符串，表示一个单词。
输出
第一行输出最长的基础上字典序最小的S1串，以及S1串为该串的单词对个数，用一个空格隔开。
第二行输出最长的基础上字典序最小的S2串，以及S2串为该串的单词对个数，用一个空格隔开。
样例输入 Copy
5
bbbaa
aacbb
bbdaa
aaaaa
bbcaa
样例输出 Copy
aa 1
aa 6
提示
对于100%的数据，2 ≤ n ≤ 50000，字符串总长不超过500000，保证单词表中所有字符都是小写字母，答案中S1串和S2串非空。

最近复习了下字典树，让后在巨巨的指导下补了这个题。
字典树在统计前缀后缀都比较好用。在构建字典树的时候，每一个点 idx 对应的 cnt [ idx ] 的数量即为以这个字母为结尾的公共前缀个数。对于这个题来说，当 cnt [ idx ] > 1 的时候即符合题目中要求的条件。
直接考虑dfs爆搜一遍，每次枚举 a ~ z 字母，如果存在儿子并且 cnt [ idx ] > 1 那么就可以扩展这个前缀，使前缀变得更长。
对于后缀是一样处理，无非是把字符串反过来，反向建一颗字典树，正是由于反向建树，所以原来要求的字典序最小，在求这个后缀的时候应该是字典序最大的。这个一开始没想明白，dfs的时候从小到大枚举字母，以为贪心找字典序最小就不需要判断长度相等的时候了，一直wa。所以
(1)长度变大的时候，理所当然的应该更新答案。
(2)长度与原来相等的时候，需要检查一下是否可以更新。
最后还要求输出单词的对数，直接套公式：n * ( n - 1 ) / 2 。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#define X first
#define Y second
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int N=500010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n;
int son[2][N][26],cnt[2][N],idx[2];
int dis[2],lc[2];
char str[N],ans[2][N],tp[2][N];

void insert(int op,char str[])
{
	if(!op)
	{
		int p=0;
		for(int i=0;str[i];i++)
		{
			int t=str[i]-'a';
			if(!son[op][p][t]) son[op][p][t]=++idx[op];
			p=son[op][p][t];
			cnt[op][p]++;
		}
	}
	else 
	{
		int p=0,l=strlen(str);
		for(int i=l-1;i>=0;i--)
		{
			int t=str[i]-'a';
			if(!son[op][p][t]) son[op][p][t]=++idx[op];
			p=son[op][p][t];
			cnt[op][p]++;
		}		
	}
}

void unit(int len,int op,int u)//更新答案
{
	dis[op]=len;
	lc[op]=cnt[op][u];
	if(!op) 
	{
		for(int i=0;i<len;i++)
			ans[op][i]=tp[op][i];
	}
	else 
	{
		for(int i=len-1,j=0;i>=0;i--,j++)
			ans[op][j]=tp[op][i];		
	}	
	ans[op][len]='\0';
}

bool check(int len,int op)//检查是否字典序更小
{
	for(int i=0,j=len-1;i<len;i++,j--)
		if(!op&&ans[op][i]>tp[op][i]) 
			return true;
		else if(!op&&ans[op][i]<tp[op][i])
			return false;
		else if(op&&ans[op][i]>tp[op][j])
			return true;
		else if(op&&ans[op][i]<tp[op][j])
			return false;
	return false;
}

void dfs(int op,int len,int u)
{
	for(int i=0;i<26;i++)
	{
		if(!son[op][u][i]) continue;
		int t=son[op][u][i];
		if(cnt[op][t]>1)
		{
			tp[op][len]=i+'a';
			dfs(op,len+1,t);
			tp[op][len]='\0';
		}
	}
	if(len>dis[op])
		unit(len,op,u);
	else if(len==dis[op]&&check(len,op))
		unit(len,op,u);
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);
	
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		scanf("%s",str);
		insert(0,str);
		insert(1,str);
	}

	dfs(0,0,0);
	dfs(1,0,0);

	printf("%s %lld\n",ans[0],1ll*lc[0]*(lc[0]-1)/2);
	printf("%s %lld\n",ans[1],1ll*lc[1]*(lc[1]-1)/2);






	return 0;
}
/*
5
bbbaa
aacbb
bbdaa
aaaaa
bbcaa
*/