树DP [ZJOI2008]骑士

问题 G: [ZJOI2008]骑士
时间限制: 1 Sec 内存限制: 162 MB
提交: 104 解决: 38
[提交][状态][讨论版]
题目描述
Z国的骑士团是一个很有势力的组织，帮会中汇聚了来自各地的精英。他们劫富济贫，惩恶扬善，受到社会各
界的赞扬。最近发生了一件可怕的事情，邪恶的Y国发动了一场针对Z国的侵略战争。战火绵延五百里，在和平环境
中安逸了数百年的Z国又怎能抵挡的住Y国的军队。于是人们把所有的希望都寄托在了骑士团的身上，就像期待有一
个真龙天子的降生，带领正义打败邪恶。骑士团是肯定具有打败邪恶势力的能力的，但是骑士们互相之间往往有一
些矛盾。每个骑士都有且仅有一个自己最厌恶的骑士（当然不是他自己），他是绝对不会与自己最厌恶的人一同出
征的。战火绵延，人民生灵涂炭，组织起一个骑士军团加入战斗刻不容缓！国王交给了你一个艰巨的任务，从所有
的骑士中选出一个骑士军团，使得军团内没有矛盾的两人（不存在一个骑士与他最痛恨的人一同被选入骑士军团的
情况），并且，使得这支骑士军团最具有战斗力。为了描述战斗力，我们将骑士按照1至N编号，给每名骑士一个战
斗力的估计，一个军团的战斗力为所有骑士的战斗力总和。

输入
第一行包含一个正整数N，描述骑士团的人数。接下来N行，每行两个正整数，按顺序描述每一名骑士的战斗力
和他最痛恨的骑士。

输出
应包含一行，包含一个整数，表示你所选出的骑士军团的战斗力。

样例输入
3
10 2
20 3
30 1
样例输出
30
提示
N ≤ 1 000 000，每名骑士的战斗力都是不大于 1 000 000的正整数。

这玩意说仇恨是单向的，但实际上它是双向的，两个骑士只能二选一。
首先假设它只有树，那么DP方程就很好想了，f[i][0]表示第i个点及其子树切不选它本身的最优方案，f[i][1]同理，树DP一下就很简单了。
但是它肯定是树套环。但是一棵树上只会有一个环。考虑把环拆掉。也就是固定砍断一条边，强迫一个端点不能选，从另一个端点开始树DP.之后交换两个点，再跑一边，就可以得出这个树套环对答案的最大贡献了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>
#define N 1000005
#define ll long long
using namespace std;
struct road{int v,next;}lu[N*2];
int read()
{
     int sum=0,f=1;char x=getchar();
     while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}
     while(x>='0'&&x<='9'){sum=(sum<<3)+(sum<<1)+x-'0';x=getchar();}
     return sum*f;
}
int n,e,st,ed,adj[N],vis[N],cut[N];ll w[N],f[N][2],ans;
void add(int u,int v){lu[e]=(road){v,adj[u]};adj[u]=e++;}
void dp(int x,int fa,int y)
{
     f[x][0]=0;vis[x]=1;
     if(x!=y)f[x][1]=w[x];
     else f[x][1]=0;
     for(int i=adj[x];~i;i=lu[i].next)
     {
         int to=lu[i].v;if(to==fa||cut[i]==-1)continue;
         dp(to,x,y);
         f[x][0]+=max(f[to][1],f[to][0]);
         f[x][1]+=f[to][0];
     }
}
void dfs(int x,int fa)
{
     vis[x]=1;
     for(int i=adj[x];~i;i=lu[i].next)
     {
         int to=lu[i].v;if(to==fa)continue;
         if(vis[to])
         {
              st=x;ed=to;
              cut[i]=cut[i^1]=-1;
              return;
         }
         dfs(to,x);
     }
}
int main()
{
    scanf("%d",&n);int x;
    memset(adj,-1,sizeof(adj));
    for(int i=1;i<=n;i++)
    {
        scanf("%lld%d",&w[i],&x);
        add(i,x);add(x,i); 
    }
    for(int i=1;i<=n;i++)
       if(!vis[i])
       {
           dfs(i,0);ll k=0;
           dp(st,0,ed);
           k=max(f[st][0],f[st][1]);
           dp(ed,0,st);
           k=max(k,max(f[ed][0],f[ed][1]));
           ans+=k;
       }
    printf("%lld\n",ans);
}