题意:
Z国的骑士团是一个很有势力的组织,帮会中汇聚了来自各地的精英。他们劫富济贫,惩恶扬善,受到社会各
界的赞扬。最近发生了一件可怕的事情,邪恶的Y国发动了一场针对Z国的侵略战争。战火绵延五百里,在和平环境
中安逸了数百年的Z国又怎能抵挡的住Y国的军队。于是人们把所有的希望都寄托在了骑士团的身上,就像期待有一
个真龙天子的降生,带领正义打败邪恶。骑士团是肯定具有打败邪恶势力的能力的,但是骑士们互相之间往往有一
些矛盾。每个骑士都有且仅有一个自己最厌恶的骑士(当然不是他自己),他是绝对不会与自己最厌恶的人一同出
征的。战火绵延,人民生灵涂炭,组织起一个骑士军团加入战斗刻不容缓!国王交给了你一个艰巨的任务,从所有
的骑士中选出一个骑士军团,使得军团内没有矛盾的两人(不存在一个骑士与他最痛恨的人一同被选入骑士军团的
情况),并且,使得这支骑士军团最具有战斗力。为了描述战斗力,我们将骑士按照1至N编号,给每名骑士一个战
斗力的估计,一个军团的战斗力为所有骑士的战斗力总和。
分析:
可以发现图是由若干个连通块组成,每个连通块都是一个基环树。
我的做法是标记一下每个连通块有没有环。
如果有环则拆环,做两遍树形DP。
代码(丑):
#include <stdio.h> #include <string.h> #include <algorithm> using namespace std; #define N 1000050 #define LL long long int head[N],to[N<<1],nxt[N<<1],cnt,n; int fa[N],root,son,rt[N],so[N],lian[N]; LL f[N],g[N],val[N]; inline void add(int u,int v){ to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt; } int find(int x){ return fa[x]==x?x:fa[x]=find(fa[x]); } void dfs3(int x,int y){ f[x]=val[x]; for(int i=head[x];i;i=nxt[i]){ if(to[i]!=y){ dfs3(to[i],x); f[x]+=g[to[i]]; g[x]+=max(f[to[i]],g[to[i]]); f[to[i]]=g[to[i]]=0; } } } void dfs1(int x,int y){ if(x!=son)f[x]=val[x]; for(int i=head[x];i;i=nxt[i]){ if(to[i]!=y){ dfs1(to[i],x); if(x!=son) f[x]+=g[to[i]]; if(to[i]==son)g[x]+=g[to[i]]; else g[x]+=max(f[to[i]],g[to[i]]); f[to[i]]=g[to[i]]=0; } } } void dfs2(int x,int y){ if(x!=root)f[x]=val[x]; for(int i=head[x];i;i=nxt[i]){ if(to[i]!=y){ dfs2(to[i],x); if(x!=root) f[x]+=g[to[i]]; if(to[i]==root)g[x]+=g[to[i]]; else g[x]+=max(g[to[i]],f[to[i]]); f[to[i]]=g[to[i]]=0; } } } int main(){ scanf("%d",&n); for(int i=1;i<=n;i++)fa[i]=i; int x; for(int i=1;i<=n;i++){ scanf("%lld%d",&val[i],&x); int di=find(i),dx=find(x); if(di!=dx){ fa[di]=dx;lian[dx]|=lian[di]; add(i,x);add(x,i); } else{ rt[di]=rt[dx]=i; so[di]=so[dx]=x; lian[di]=lian[dx]=1; } } LL ans=0; for(int i=1;i<=n;i++){ if(fa[i]==i){ if(!lian[i]){ dfs3(i,0); ans+=max(f[i],g[i]); }else{ root=rt[i],son=so[i]; dfs1(root,0); LL now=max(f[root],g[root]); f[root]=g[root]=0; dfs2(son,0); ans+=max(max(now,f[son]),g[son]); } } } printf("%lld\n",ans); }
原文地址:https://www.cnblogs.com/suika/p/8476370.html
时间: 2024-11-05 22:42:50