您的位置:首页 > 其它

Codeforces Similar Words [trie树+树形dp]

2017-09-22 21:48 387 查看
题意:给你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
;
int ru
,dp
[2],ch
[30],mark
;
char a
;
int tot=0;
int newnode()
{
tot++;
for(int i=0;i<30;i++)
ch[tot][i]=-1;
return tot;
}
string A
;
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);
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: