标签:树形DP,环套树
Description
Z国的骑士团是一个很有势力的组织,帮会中汇聚了来自各地的精英。他们劫富济贫,惩恶扬善,受到社会各
界的赞扬。最近发生了一件可怕的事情,邪恶的Y国发动了一场针对Z国的侵略战争。战火绵延五百里,在和平环境
中安逸了数百年的Z国又怎能抵挡的住Y国的军队。于是人们把所有的希望都寄托在了骑士团的身上,就像期待有一
个真龙天子的降生,带领正义打败邪恶。骑士团是肯定具有打败邪恶势力的能力的,但是骑士们互相之间往往有一
些矛盾。每个骑士都有且仅有一个自己最厌恶的骑士(当然不是他自己),他是绝对不会与自己最厌恶的人一同出
征的。战火绵延,人民生灵涂炭,组织起一个骑士军团加入战斗刻不容缓!国王交给了你一个艰巨的任务,从所有
的骑士中选出一个骑士军团,使得军团内没有矛盾的两人(不存在一个骑士与他最痛恨的人一同被选入骑士军团的
情况),并且,使得这支骑士军团最具有战斗力。为了描述战斗力,我们将骑士按照1至N编号,给每名骑士一个战
斗力的估计,一个军团的战斗力为所有骑士的战斗力总和。
Input
第一行包含一个正整数N,描述骑士团的人数。接下来N行,每行两个正整数,按顺序描述每一名骑士的战斗力
和他最痛恨的骑士。
Output
应包含一行,包含一个整数,表示你所选出的骑士军团的战斗力。
Sample Input
3
10 2
20 3
30 1
Sample Output
30
HINT
N ≤ 1 000 000,每名骑士的战斗力都是不大于 1 000 000的正整数。
题意:许多点构成的点集,选取权值总和最大的一个点集,保证其中任意两点没有直接连边。
本题实质上是许多无向边构成的环,整个图就是无向环套树森林,先dfs找环然后破环为链,选取根节点做树形DP,取结果最大值就可以了
Code
#include<iostream> #include<iomanip> #include<cstdio> #include<cstdlib> #include<algorithm> #include<cstring> #define rep(i,a,b) for(int i=a;i<=b;i++) #define dep(i,a,b) for(int i=a;i>=b;i--) #define mem(x,num) memset(x,num,sizeof x) #define reg(x) for(int i=head[x];i;i=e[i].next) #define LL long long using namespace std; inline LL read() { LL f=1,x=0;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } const int maxn=1e6+6; int head[maxn],cnt=1,b,a[maxn],visit[maxn],r,u,v; LL f[maxn],g[maxn],ans,n,x; struct edge{int to,next;}e[maxn<<1]; inline void dfs(int x,int from) { visit[x]=1; reg(x){ if((i^1)==from)continue; if(visit[e[i].to]){ u=x; v=e[i].to; r=i; continue; } dfs(e[i].to,i); } } void dp(int x,int from,int ban) { f[x]=a[x];g[x]=0; reg(x){ if(i==ban||(i^1)==ban)continue; if((i^1)==from)continue; dp(e[i].to,i,ban); f[x]+=g[e[i].to]; g[x]+=max(f[e[i].to],g[e[i].to]); } } int main() { n=read(); rep(i,1,n){ a[i]=read(),x=read(); e[++cnt]=(edge){i,head[x]};head[x]=cnt; e[++cnt]=(edge){x,head[i]};head[i]=cnt; } rep(i,1,n) if(!visit[i]){ dfs(i,0); dp(u,0,r); LL temp=g[u]; dp(v,0,r); temp=max(temp,g[v]); ans+=temp; } printf("%lld\n",ans); return 0; }