词韵（字典树+DP）

                                                词韵

                                                                时间限制: 2 Sec  内存限制: 128 MB
                                                                            提交: 167  解决: 29
                                                                [提交] [状态] [讨论版] [命题人:admin]

题目描述

Adrian 很喜欢诗歌中的韵。他认为，两个单词押韵当且仅当它们的最长公共 后缀的长度至少是其中较长单词的长度减一。也就是说，单词 A 与单词 B 押韵 当且仅当 LCS(A, B) ≥ max(|A|, |B|) – 1。（其中 LCS 是最长公共后缀 longest common suffix 的缩写）
现在，Adrian 得到了 N 个单词。他想从中选出尽可能多的单词，要求它们能 组成一个单词序列，使得单词序列中任何两个相邻单词是押韵的。

 

输入

第一行是一个整数N。
接下来N行，每行一个由小写英文字母组成的字符串，表示每个单词。所有单词互不相同。

 

输出

输出一行，为一个整数，表示最长单词序列的长度。

 

样例输入

5
ask
psk
k
krafna
sk

样例输出

4

提示

一种最长单词序列是 ask-psk-sk-k。
30%的测试数据：1 ≤ N ≤ 20，所有单词长度之和不超过 3 000。
100%的测试数据：1 ≤ N ≤ 500 000，所有单词长度之和不超过 3 000 000。

                                                                                          [提交]      [状态]

按照题意，如果两个单词是押韵的那么这两个单词的长度之差不超过一（前面的比后面的长或者后面的比前面的长）。所以将单词逆置， 后缀变前缀， 建立字典树。

然后分两种情况去考虑下：①就是寻找最长的单链 ② 找2个最长的子链（最长子链和次长子链）拼接。

以下所说的节点都是储存单词结尾的节点 这样的点才是有用的点

对于叶子节点 他得贡献肯定的自己本身的单词的出现次数也就是valu[i]，对于他的父亲，假如只有一个儿子的话，那么他的到的贡献肯定是等于valu[儿子]+valu[自己] ,如果有多个儿子的话，可选择2个拥有最长链的儿子拼成一个。

valu[父亲]=valu[父亲]+valu[max]+valu[max1]。

大概的意思可以组成一个 abc->ab->a->ac->acd 这样的一个链也是满足题意的。最后就变成了单链和合并2个孩子求一个max即可。

 

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
char s[maxn];
int head[maxn];
int valu[maxn];
int dp[maxn],dp1[maxn];
int cnt;
struct node{
	char c;
	int to;
}edge[maxn];

void add(int root,char s)
{
    edge[++cnt].c=s;
    edge[cnt].to=head[root];
    head[root]=cnt;

}

void built()//链式前向星的建立字典树
{

	int root=0;
    int len = strlen(s);
    for(int i=0;i<len;i++)
	{

		int flag=1;
        for(int j=head[root];j!=-1;j=edge[j].to)
		{

			if(s[i] == edge[j].c)
			{

				flag=0;
				root=j;
				break;
			}
		}
		if(flag)
		{
			add(root,s[i]);
			root=cnt;

		}

	}

	valu[root]++;//单词的数量

}

void dfs(int root)
{//叶子节点就是valu[i]

	int Valu[2]={0,0};
	int flag=0;//标记点
	for(int i=head[root];i!=-1;i=edge[i].to)//根节点才会走着 叶子节点不会走这
	{

		dfs(i);//递归到最下
		//Valu数组的作用就是求root的孙子辈最长的一个链 然后用flag串起来他所有儿子
		int Mc=dp1[i] - valu[i];
		if(Mc > Valu[1])
		{

			Valu[0]=Valu[1];//两条链 一条最大一条次大
			Valu[1]=Mc;//单链

		}

		else if(Mc > Valu[0])
			Valu[0] = Mc;
		flag = flag + valu[i];

	}


	flag = flag +valu[root];
	if(Valu[0]>0)//连接2个孩子
		dp[root] = Valu[0] + Valu[1] + flag;

	if(valu[root])//如果当前点是一个单词的结束； 如果是根的话  dp[root]就等于 valu[root]
		dp1[root]= flag + Valu[1];//不是的话就加上最长的那个链

}

int main()
{

	memset(head,-1,sizeof(head));
	int n;
	scanf("%d",&n);
	while(n--)
	{

		scanf("%s",s);
		int len = strlen(s);
		for(int i=0;i<len/2;i++)
			swap(s[i],s[len-i-1]);//单词逆序
		built();
	}
	dfs(0);
	int Ans=0;
	for(int i=0;i<=cnt;i++)
		Ans=max(Ans,max(dp[i],dp1[i]));

	printf("%d\n",Ans);

	return 0;

}