（bzoj 1040 [ZJOI2008]骑士）<基环树森林dp>

传送门

Solution

• 先贴一个讲的很好的博客
• 基环树是n个点n条边的图，删除环上任意一条边即可成为树
• 如果A与B有矛盾，则在A，B之间连一条无向边，最终会建成一颗基环树
• 题意是要求任意相邻的两点不能同时取，如果是一棵树的话，就是经典的“没有上司的舞会”问题。那么如何将基环树转化为树呢，只要将环上的任意一条边删去即可。而无根树的性质就是可以任意指定根而且不会影响答案，因此用vis判重
• DP方程：f[i]表示选i这个点时，在它的子树中可选取的最大权值
  g[i]表示不选i这个点时，在它的子树中可选取的最大权值
  $$f[i]=\sum_{v=e[i].to}g[v]$$
  $$g[i]=\sum_{v=e[i].to}\max(g[v],f[v])$$
• 假设现在环上的一条边已经被删去，将它命名为u->v的边，则以u为根跑一遍DP，再以v为根跑一遍DP，比较两次答案的大小，取max加入答案，这样无论是DP时还是枚举根时都可以满足题意

Code

// by spli
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define LL long long
using namespace std;

const int N=1000010;
int n;
struct node{
    int to,nxt;
}e[N<<1];int head[N],cnt;
int val[N];
bool vis[N];
LL f[N],g[N];
int s,t,p;

void add(int f,int t){
    e[cnt]=(node){t,head[f]};
    head[f]=cnt++;
}

void dfs(int u,int fa){
    vis[u]=1;
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa) continue;
        if(!vis[v]) dfs(v,u);
        else{
            s=u,t=v;
            p=i;
        }
    }
}

void DP(int u,int fa){
    f[u]=val[u];g[u]=0;
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v==fa) continue;
        if(i==p||i==(p^1)) continue;
        DP(v,u);
        f[u]+=g[v];
        g[u]+=max(g[v],f[v]);
    }
}

int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    int c,x;
    for(int i=1;i<=n;++i){
        scanf("%d%d",&c,&x);
        add(i,x);add(x,i);
        val[i]=c;
    }
    LL tmp=0,ans=0;
    for(int i=1;i<=n;++i){
        if(vis[i]) continue;
        dfs(i,-1);
        DP(s,-1);
        tmp=g[s];
        DP(t,-1);
        ans+=max(tmp,g[t]);
    }
    cout<<ans;
    return 0;
}