[P2607][ZJOI2008]骑士(基环树)

边数等于点数,是基环树。考虑dp,如果是树的话,转移方程为dp[u][1](选择u)=w[u]+Σdp[v][0],v为u的子节点; dp[u][0](不选择u)=Σmax(dp[v][0],dp[v][1])。然后基环树只要找到环后,断开环上的一条边就成了树。断开的边上两端点不能同时选择,所以对于每棵基环树,我们求两边dp,分别限制两个端点必须不选,然后取最大的结果。每棵树答案相加即为最终答案。

#include
#include
#include
#define ll long long
using namespace std;
const int N=1e6+10;
struct edge{
	int y,next;
}data[N<<1];
int n,w1[N],num,num1,h[N],dfn[N],utmp,vtmp,idtmp;
ll ans,ans1,ans2,dp[N][3];
inline int read(){
	int x=0;char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x;
}
inline void addedge(int u,int v){
	data[++num].y=v,data[num].next=h[u],h[u]=num;
	data[++num].y=u,data[num].next=h[v],h[v]=num;
}
void dfs1(int u,int fa){
	dfn[u]=++num1;
	for(int i=h[u],v;i!=-1;i=data[i].next){
		v=data[i].y;if(fa==v)continue;
		if(dfn[v]){
			if(dfn[v]b?a:b;}
void getdp(int u,int fa){
	dp[u][1]=w1[u],dp[u][0]=0;
	for(int i=h[u],v;i!=-1;i=data[i].next){
		v=data[i].y;if(fa==v||idtmp==i||(idtmp^1)==i)continue;
		getdp(v,u);dp[u][1]+=dp[v][0],dp[u][0]+=max1(dp[v][1],dp[v][0]);
	}
}
int main(){
	n=read();num=-1;num1=0;
	memset(h,-1,sizeof h),memset(dfn,0,sizeof dfn);
	for(int i=1,v;i<=n;++i)w1[i]=read(),v=read(),addedge(v,i);
	ans=0;
	for(int i=1;i<=n;++i)
		if(!dfn[i]){
			dfs1(i,0);
			getdp(utmp,0),ans1=dp[utmp][0];
			getdp(vtmp,0),ans2=dp[vtmp][0];
			ans+=max1(ans1,ans2);
		}
	printf("%lld",ans);
	return 0;
} 

 

你可能感兴趣的:(基环树)