[bzoj1040][ZJOI2008]骑士（树上dp）

【题目链接】http://www.lydsy.com/JudgeOnline/problem.php?id=1040
【题目大意】每个人有战斗力和一个痛恨的人，他不能与那个人一起出征，求最大战斗力总和
【解题思路】整体的图由一个或多个联通块构成。每个联通块点数等于边数。由于每个点至少连一条边所以这是一个环套树（在一棵树上加一条边构成的图）。dfs随意拆环，按被拆边两个端点的选取情况进行两次dp择优作为此联通块答案。
【呆马】

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<iostream>
#define ll long long
const int N=1e6+1;
using namespace std;
struct st{int to,next,op;} e[N<<1];
int n,i,x,p,q,ed1,ed2,cnt,fi[N],v[N];
ll t,ans,f[N][2];
bool vis[N];
void add(int x,int y)
{
    e[++cnt].to=y; e[cnt].next=fi[x]; e[cnt].op=cnt+1; fi[x]=cnt;
    e[++cnt].to=x; e[cnt].next=fi[y]; e[cnt].op=cnt-1; fi[y]=cnt;
}

void dfs(int x,int fa)
{
    vis[x]=1;
    for (int i=fi[x];i;i=e[i].next)
        if (e[i].to!=fa)
        {
            if (vis[e[i].to]) p=x,q=e[i].to,ed1=i,ed2=e[i].op;
            else dfs(e[i].to,x);
        }
}

void dp(int x,int fa)
{
    f[x][0]=0;
    f[x][1]=v[x];
    for (int i=fi[x];i;i=e[i].next)
        if (i!=ed1 && i!=ed2 && e[i].to!=fa)
        {
            int y=e[i].to;
            dp(y,x);
            f[x][0]+=max(f[y][0],f[y][1]);
            f[x][1]+=f[y][0];
        }
}

int main()
{
        scanf("%d\n",&n);
        for (i=1;i<=n;i++)
        {
            scanf("%d%d\n",&v[i],&x);
            add(i,x);
        }
        for (i=1;i<=n;i++)
            if (!vis[i])
            {
                p=q=0;
                dfs(i,0);
                dp(p,0);
                t=f[p][0];
                dp(q,0);
                ans+=max(t,f[q][0]);
            }
        printf("%lld",ans);
}