Codeforces Similar Words [trie树+树形dp]

7 篇文章 0 订阅
5 篇文章 0 订阅

题意:给你n个串,定义串A与串B相似为:A删掉第一个字符与B一样或者B删掉第一个字符与A一样,设X串中只能包含给出的n个串的所有前缀,并且不能出现相似的两个串,问S串最大为多少。

题解:我们对当前串和删掉其第一个字符的串连边(删掉一个字符的串假如不是某个串的前缀则不连边),这时候肯定生成了森林,这时候问题就转化为,在树中取不相邻节点的个数的最大值,那我们对每棵树dp取或者不取的情况即可。我们可以用tire树作为前缀的hash,建立森林。

注意:重边要去重,不然T。

AC代码:

#include<stdio.h>  
#include<map>  
#include<queue>  
#include<string.h>
#include<algorithm>  
#include<vector>
#include<string>
#define N 1000005
using namespace std;  
typedef long long ll;
vector<int>vt[N];
int ru[N],dp[N][2],ch[N][30],mark[N];
char a[N];
int tot=0;
int newnode()
{
	tot++;
	for(int i=0;i<30;i++)
		ch[tot][i]=-1;
	return tot;
}
string A[N];
void dfs(int u)
{
	mark[u]=1;
	dp[u][0]=0;
	dp[u][1]=1;
	for(int i=0;i<vt[u].size();i++)
	{
		int to=vt[u][i];
		if(mark[to])continue;
		dfs(to);
		dp[u][0]+=max(dp[to][1],dp[to][0]);
		dp[u][1]+=dp[to][0];
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		tot=0;
		int n;
		scanf("%d",&n);
		for(int i=0;i<30;i++)ch[0][i]=-1;
		for(int i=0;i<n;i++)
		{
			scanf("%s",a);
			int l=strlen(a);
			A[i]=a;
			int now=0;
			for(int j=0;j<l;j++)
			{
				if(ch[now][a[j]-'a']==-1)ch[now][a[j]-'a']=newnode();
				now=ch[now][a[j]-'a'];
			}
		}
		for(int i=1;i<=tot;i++)
			vt[i].clear(),ru[i]=0,mark[i]=0;
		for(int i=0;i<n;i++)
		{
			int len=A[i].size();
			int flag=1,now1=0,now2=0;
			for(int j=0;j<len;j++)
			{
				now1=ch[now1][A[i][j]-'a'];
				if(j>0)
				{
					now2=ch[now2][A[i][j]-'a'];
					if(now2==-1)flag=0;
					if(flag)vt[now2].push_back(now1),ru[now1]++;
				}
			}
		}
		int ans=0;
		for(int i=1;i<=tot;i++)
			if(ru[i]==0&&mark[i]==0)
			{
				dfs(i);
				ans+=max(dp[i][0],dp[i][1]);
			}
		printf("%d\n",ans);
	}
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值