[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);
}