bzoj1040: [ZJOI2008]骑士（树形dp）

Problem

每个人有一个战斗力值，有一个敌对对象。
每个人不能和他的敌对对象在一个军团。
问选出来的人组成一个军团且不存在敌对对象的最大战斗力和为多少

Solution

显然会出环（$n$ 个点 $n$ 条边），假如不考虑环的出现，就是相邻两点不能同时选的树形 $dp$ 。

加上环呢？
对应本题，会发现之多有一个环…读题可知…
因此 $dfs$ 找到非树边
然后对于这条边，分为不选左点右点随意、不选右点左点随意两种情况取最大值

Code

#include <cstdio>
#include <algorithm>
using namespace std;
#define ll long long
#define N 1000010
int n,num=1,h[N],L,R,E,vis[N];
ll a[N],ans=0;
struct node{int to,next;}mp[N<<1];
inline void insert(int x,int y){
    mp[++num].to=y;mp[num].next=h[x];h[x]=num;
    mp[++num].to=x;mp[num].next=h[y];h[y]=num;
}
int find(int x,int f){
    vis[x]=1;
    for(int i=h[x];i;i=mp[i].next){
        int y=mp[i].to;if(y==f) continue;
        if(vis[y]){
            L=x,R=y,E=i;
            continue;
        }
        find(y,x);
    }
}
ll dp[N][2];//0 不选；1 选 
void dfs(int x,int f,int e){
    dp[x][0]=0;dp[x][1]=a[x];
    for(int i=h[x];i;i=mp[i].next){
        int y=mp[i].to;if(y==f || i==e || i==(e^1)) continue;
        dfs(y,x,e);
        dp[x][0]+=max(dp[y][0],dp[y][1]);
        dp[x][1]+=dp[y][0];
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;scanf("%lld%d",&a[i],&x);
        insert(i,x);
    }
    for(int i=1;i<=n;i++)
        if(!vis[i]){
            find(i,0);
            dfs(L,0,E);ll t1=dp[L][0];
            dfs(R,0,E);ll t2=dp[R][0];
            ans+=max(t1,t2);
        }
    printf("%lld\n",ans);
    return 0;
}