BZOJ 1040: [ZJOI2008]骑士

锵锵锵！！！骑士驾到！！！
这道题tree_dp必须出场~(≧▽≦)/~
给出你一张图（一定要记得是可能不连通，所以可能有好几个环）
要求有边相连的点不能同时选，每个点都有权值，请选出一些点使得选出的点组成的子图权值和最大
那么有环路就妨碍了我们直接使用tree_dp（好讨厌(ノω<。)ノ))☆.。），所以我们要把环断开，然后在进行tree_dp

---

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=1000000+5;
int n,cnt;
int s,e,edge;
long long f[maxn][2],ans;
int w[maxn],hd[maxn],to[2*maxn],nxt[2*maxn],vis[maxn];
void add(int x,int y){
    to[cnt]=y;
    nxt[cnt]=hd[x];
    hd[x]=cnt++;
}
void circle(int root,int sf){
    vis[root]=1;
    for(int i=hd[root];i!=-1;i=nxt[i]){
        if((i^1)==sf)
            continue;
        if(vis[to[i]])
            s=root,e=to[i],edge=i;
        else
            circle(to[i],i);
    }
}
void tree_dp(int root,int sf){
    f[root][0]=0;
    f[root][1]=w[root];
    for(int i=hd[root];i!=-1;i=nxt[i]){
        if((i^1)==sf||i==edge||(i^1)==edge)
            continue;//如果当前边与上一条走过来的边相反，那么不能再走回去是不是 
        tree_dp(to[i],i);
        f[root][0]+=max(f[to[i]][0],f[to[i]][1]);
        f[root][1]+=f[to[i]][0];
    }
}
int main(){
    scanf("%d",&n);
    ans=0,cnt=0,memset(hd,-1,sizeof(hd)),memset(vis,0,sizeof(vis));
    for(int i=1,x;i<=n;i++)
        scanf("%d%d",&w[i],&x),add(i,x),add(x,i);
    for(int i=1;i<=n;i++){
        if(vis[i])
            continue;
        circle(i,-1);
        tree_dp(s,-1);
        long long ans1=f[s][0];
        tree_dp(e,-1);
        ans+=max(f[e][0],ans1);//一定是+=因为可能有好几个环
    }
    cout<<ans<<endl;
    return 0;
}

---

by >o< neighthorn