[BZOJ1791][IOI2008]Island岛屿(环套树DP)

同NOI2013快餐店（NOI出原题？），下面代码由于BZOJ栈空间过小会RE。

大致是对每个连通块找到环，在所有内向树做一遍DP，再在环上做两遍前缀和优化的DP。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
#define For(i,x) for (int i=h[x],k; i; i=nxt[i])
typedef long long ll;
using namespace std;

const int N=1000010;
ll ans,res,f[N],v1[N],v2[N],u1[N],u2[N];
int dfn[N],fa[N],pre[N],h[N],a[N],b[N];
int n,x,w,cnt,tim,tot,to[N<<1],val[N<<1],nxt[N<<1],d[N];
void add(int u,int v,int w){ to[++cnt]=v; val[cnt]=w; nxt[cnt]=h[u]; h[u]=cnt; }

void dfs(int x){
    dfn[x]=++tim;
    For(i,x) if ((k=to[i])!=fa[x]){
        if (!dfn[k]) fa[k]=x,pre[k]=val[i],dfs(k);
        else if (dfn[k]>dfn[x]){
            for (int t=k; t!=x; t=fa[t]) a[++tot]=t,d[t]=1,b[tot]=pre[t];
            a[++tot]=x; d[x]=1; b[tot]=val[i];
        }
    }
}

void Dfs(int x,int fa){
    For(i,x) if ((k=to[i])!=fa && !d[k])
        Dfs(k,x),ans=max(ans,f[x]+f[k]+val[i]),f[x]=max(f[x],f[k]+val[i]);
}

void DP(int x){
    tot=0; ans=0; dfs(x); ll sm=0,mx=0;
    rep(i,1,tot) Dfs(a[i],0);
    rep(i,1,tot) d[a[i]]=0;
    rep(i,0,tot+1) u1[i]=u2[i]=v1[i]=v2[i]=0;
    rep(i,1,tot){
        sm+=b[i-1]; u1[i]=max(u1[i-1],f[a[i]]+sm);
        v1[i]=max(v1[i-1],f[a[i]]+sm+mx); mx=max(mx,f[a[i]]-sm);
        ans=max(ans,v1[i]);
    }
    int tmp=b[tot]; sm=mx=b[tot]=0;
    for (int i=tot; i; i--){
        sm+=b[i]; u2[i]=max(u2[i+1],f[a[i]]+sm);
        v2[i]=max(v2[i+1],f[a[i]]+sm+mx); mx=max(mx,f[a[i]]-sm);
        ans=max(ans,v2[i]);
    }
    rep(i,1,tot-1) ans=max(ans,u1[i]+u2[i+1]+tmp);
    res+=ans;
}

int main(){
    freopen("bzoj1791.in","r",stdin);
    freopen("bzoj1791.out","w",stdout);
    scanf("%d",&n);
    rep(i,1,n) scanf("%d%d",&x,&w),add(x,i,w),add(i,x,w);
    rep(i,1,n) if (!dfn[i]) DP(i);
    printf("%lld\n",res);
    return 0;
}