bzoj 1040 [ZJOI2008]骑士 基环树 dp

这是一个基环树森林。
对于每块，对于环上的每一棵树设f[i][0/1]表示选/不选点i，i子树中的最大收益。
对于每个环设g[i][0/1][0/1]表示第一个点选/不选，第i个点选/不选的最大收益。
然后把每个基环树的收益加和就是答案。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define N 1100000
int n,tim,top;
ll f[N][2],val[N],st[N],g[N][2][2],ans;
int nex[N],vis[N],inc[N];
vector<int>son[N];
void dfs(int x)
{
    vis[x]=1;f[x][1]=val[x];
    for(int i=0,t;i<son[x].size();i++)
        if(!inc[t=son[x][i]])
        {
            dfs(t);
            f[x][0]+=max(f[t][0],f[t][1]);
            f[x][1]+=f[t][0];
        }
}
ll calc()
{
    g[1][1][1]=f[st[1]][1];
    g[1][0][0]=f[st[1]][0];
    for(int i=2;i<=top;i++)
        for(int j=0;j<=1;j++)
        {
            g[i][j][0]=max(g[i-1][j][0],g[i-1][j][1])+f[st[i]][0];
            g[i][j][1]=g[i-1][j][0]+f[st[i]][1];
        }
    return max(max(g[top][1][0],g[top][0][1]),g[top][0][0]);
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld%d",&val[i],&nex[i]);
        son[nex[i]].push_back(i);
    }   
    for(int i=1;i<=n;i++)
        if(!vis[i])
        {
            int t=i;tim++;
            for(;!vis[t];t=nex[t])vis[t]=tim;
            if(vis[t]==tim)
            {
                inc[t]=1;
                for(int j=nex[t];j!=t;j=nex[j])inc[j]=1;
            }
        }
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
        if(!vis[i]&&inc[i])
        {
            st[top=1]=i;dfs(i);
            for(int j=nex[i];j!=i;j=nex[j])
                dfs(j),st[++top]=j;
            ans+=calc();
        }
    printf("%lld\n",ans);
    return 0;
}