BZOJ 1040：[ZJOI2008]骑士（环套外向树+树形DP）

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long LL;
const int N=1000010;
vector<int> v[N],id[N];
LL g[N],f[N],ans;
int ban,n,c[N],x,vis[N],root,_root,idn=0;
void dfs(int x,int fx){  
    vis[x]=1;  
    for(int i=0;i<v[x].size();i++){
        int y=v[x][i]; 
        if(y==fx)continue;  
        if(!vis[y])dfs(y,x);  
        else{root=y;_root=x;ban=id[x][i];}  
    }  
}  
void TreeDP(int x,int fx){
    f[x]=c[x],g[x]=0;
    for(int i=0;i<v[x].size();i++){
        int y=v[x][i];
        if(y==fx)continue;
        if(id[x][i]==ban||id[x][i]==(ban^1))continue;
        TreeDP(y,x);
        f[x]+=g[y]; g[x]+=max(f[y],g[y]);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&c[i],&x);
        v[x].push_back(i); 
        id[x].push_back(idn++);
        v[i].push_back(x);
        id[i].push_back(idn++);
    }
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            dfs(i,-1);
            TreeDP(root,-1);
            LL tmp=g[root];
            TreeDP(_root,-1);
            tmp=max(tmp,g[_root]);
            ans+=tmp;
        }
    }printf("%lld\n",ans);
    return 0;
}