Description
小A非常喜欢所有押韵的东西,他认为两个单词押韵当且仅当他们的公共后缀的长度和两个单词中最长的单词的长度相等,或者是最长的单词的长度减一。也就是说LCS(A,B)>=max(|A|,|B|)-1。
有一天,小A读了一个有N个单词的小故事,他想知道,如果挑选一些故事里出现的单词组成一个新的单词序列,能组成的最长的满足以下条件的单词序列的长度是多少:单词序列中任意相邻的两个单词都押韵。(每个单词最多只能用一次)
Input
第一行包含一个正整数N(1<=N<=500000),表示单词的个数。
接下来N行,每行一个由英文小写字母构成的单词,表示原故事里的单词。输入保证所有单词都不一样,并且他们的总长度不超过3,000,000。
Output
输出满足条件的最长单词序列的长度。
Sample Input
输入1:
4
honi
toni
oni
ovi
输入2:
5
k
ask
psk
krafna
sk
输入3:
5
pas
kompas
stas
s
nemarime
Sample Output
输出1:
3
输出2:
4
输出3:
1
Data Constraint
有30%的数据,有N<=18。
赛时
第一眼以为是用AC自动机自动AC,后面才发现一个字典树就搞定了,随后答案的方程推错。最后就是爆栈见(xian)了祖宗,不过竟然还有40分,香香的。
正解
先用公共后缀建树,之后考虑答案的统计,对于一个节点u(假设有值)那么它的贡献就是它儿子贡献中最大的贡献(这里的贡献包含儿子本身,所以要-1)加上它的儿子节点数以及自己,之后考虑对于u的答案应该是儿子中最大,次大的贡献(同理要-1)与儿子数(以及自己)的和。
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=2000000;
int f[maxn][26],d[maxn],g[maxn];
int n,t,x,y,z,ans,cnt;
char s[maxn];
int main(){
freopen("rhyme.in","r",stdin);
freopen("rhyme.out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%s",s+1);
x=0;
t=strlen(s+1);
for (int j=t;j>=1;j--){
if (!f[x][s[j]-'a']) f[x][s[j]-'a']=++cnt;
x=f[x][s[j]-'a'];
}
d[x]++;
//插入
}
for(int x=cnt;x>=0;x--){
int t=0,k=0;
g[x]=d[x];
for(int j=0;j<26;j++){
if(!d[f[x][j]])continue;
g[x]++;
if(t<g[f[x][j]]-1) k=t,t=g[f[x][j]]-1;
else if(k<g[f[x][j]]-1) k=g[f[x][j]]-1;
}
if(k) ans=max(ans,t+k+g[x]);
g[x]+=t;
ans=max(ans,g[x]);
}
printf("%d\n",ans);
}