bzoj1040 骑士 树形dp

       类仙人掌的树形dp。对于一个联通块，一定是由一个环和若干棵树构成的。对于一个环，破环为链，设破开的点为u,v，假设u不选做一次，再假设v不选做一次。两次取较大值累加入答案即可。

AC代码如下：

#include<iostream>
#include<cstdio>
#define N 1000005
#define ll long long
using namespace std;

int n,tot=1,tx,ty,tz,a[N],fst[N],pnt[N<<1],nxt[N<<1];
ll f[N][2]; bool vis[N];
void add(int aa,int bb){
	pnt[++tot]=bb; nxt[tot]=fst[aa]; fst[aa]=tot;
}
void dfs(int x,int fa){
	int p; vis[x]=1;
	for (p=fst[x]; p; p=nxt[p]){
		int y=pnt[p]; if (y==fa) continue;
		if (vis[y]){ tx=x; ty=y; tz=p; continue; }
		dfs(y,x);
	}
}
void treedp(int x,int fa,int edg){
	f[x][0]=0; f[x][1]=a[x]; int p;
	for (p=fst[x]; p; p=nxt[p]){
		if (p==edg || (p^1)==edg) continue;
		int y=pnt[p]; if (y==fa) continue;
		treedp(y,x,edg);
		f[x][1]+=f[y][0]; f[x][0]+=max(f[y][0],f[y][1]);
	}
}
int main(){
	scanf("%d",&n); int i; ll ans=0;
	for (i=1; i<=n; i++){
		int x; scanf("%d%d",&a[i],&x);
		add(i,x); add(x,i);
	}
	for (i=1; i<=n; i++) if (!vis[i]){
		dfs(i,0); treedp(tx,0,tz); ll tmp=f[tx][0];
		treedp(ty,0,tz); tmp=max(tmp,f[ty][0]); ans+=tmp;
	}
	printf("%lld\n",ans);
	return 0;
}

by lych

2015.12.26