BZOJ1040 [ZJOI2008]骑士[基环树DP]

实际上就是一般基环树DP套路性做法，断环成树，见过一题就知道了，具体可以看上一篇。这题基本一致。。。也是考虑断开后两点的影响关系分两种讨论。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define mst(x) memset(x,0,sizeof x)
#define dbg(x) cerr << #x << " = " << x <<endl
#define dbg2(x,y) cerr<< #x <<" = "<< x <<"  "<< #y <<" = "<< y <<endl
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> pii;
template<typename T>inline T _min(T A,T B){return A<B?A:B;}
template<typename T>inline T _max(T A,T B){return A>B?A:B;}
template<typename T>inline char MIN(T&A,T B){return A>B?(A=B,1):0;}
template<typename T>inline char MAX(T&A,T B){return A<B?(A=B,1):0;}
template<typename T>inline void _swap(T&A,T&B){A^=B^=A^=B;}
template<typename T>inline T read(T&x){
    x=0;int f=0;char c;while(!isdigit(c=getchar()))if(c=='-')f=1;
    while(isdigit(c))x=x*10+(c&15),c=getchar();return f?x=-x:x;
}
const int N=1e6+7;
struct thxorz{
    int head[N],nxt[N<<1],to[N<<1],tot;
    thxorz(){tot=1;}
    inline void add(int x,int y){
        to[++tot]=y,nxt[tot]=head[x],head[x]=tot;
        to[++tot]=x,nxt[tot]=head[y],head[y]=tot;
    }
}G;
int val[N];
int n;
#define y G.to[j]
int vis[N],rt,to,ban;
void dfs(int x){
    vis[x]=1;
    for(register int j=G.head[x];j;j=G.nxt[j])if(!(j&1)){
        if(vis[y])return rt=x,to=y,ban=j,void();
        else dfs(y);
    }
}
ll f[N][2],res,ans;
void dp1(int x,int fa){
    vis[x]=1,f[x][1]=val[x];
    for(register int j=G.head[x];j;j=G.nxt[j])if(y^fa&&j^ban&&j^(ban^1)){
        dp1(y,x);
        f[x][0]+=_max(f[y][0],f[y][1]);
        f[x][1]+=f[y][0];
    }
}
void dp2(int x,int fa){
    f[x][1]=val[x],f[x][0]=0;
    for(register int j=G.head[x];j;j=G.nxt[j])if(y^fa&&j^ban&&j^(ban^1)){
        dp2(y,x);
        f[x][0]+=_max(f[y][0],f[y][1]);
        f[x][1]+=f[y][0];
    }
    if(x==to)f[x][1]=0;
}
#undef y
int main(){//freopen("test.in","r",stdin);//freopen("test.ans","w",stdout);
    read(n);
    for(register int i=1,x;i<=n;++i)read(val[i]),read(x),G.add(i,x);
    for(register int i=1;i<=n;++i)if(!vis[i]){
        dfs(i);dp1(rt,0),res=f[rt][0];
        dp2(rt,0),MAX(res,f[rt][1]),ans+=res;
    }
    printf("%lld
",ans);
    return 0;
}