2018.11.06 bzoj1040: [ZJOI2008]骑士（树形dp）

传送门
由题可知给出的是基环森林。
因此对于每个基环森林找到环断开$dp$两次就行了。
代码：

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    int ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
    return ans;
}
const int N=1e6+5;
typedef long long ll;
int ban,n,first[N],cnt=-1;
bool vis[N],flag;
ll w[N],ans=0,f[N][2],tmp;
struct edge{int v,next;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;}
inline void dfs(int p,int pre){
    f[p][0]=0,f[p][1]=w[p];
    for(int i=first[p];~i;i=e[i].next){
        int v=e[i].v;
        if(i==ban||(i^1)==ban||(i^1)==pre)continue;
        dfs(v,i),f[p][1]+=f[v][0],f[p][0]+=max(f[v][0],f[v][1]);
    }
}
inline void solve(int p,int pre){
    vis[p]=1;
    for(int i=first[p];~i;i=e[i].next){
        int v=e[i].v;
        if(pre==(i^1))continue;
        if(vis[v]){if(!flag)ban=i,dfs(p,-1),tmp=f[p][0],dfs(v,-1),tmp=max(tmp,f[v][0]),ans+=tmp,flag=1;continue;}
        solve(v,i);
    }
}
int main(){
    memset(first,-1,sizeof(first)),n=read();
    for(int u=1,v;u<=n;++u)w[u]=read(),v=read(),add(u,v),add(v,u);
    for(int i=1;i<=n;++i)if(!vis[i])flag=0,solve(i,-1);
    cout<<ans;
    return 0;
}