[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<cstdio>
#include<cstring>
#include<algorithm>
#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]<dfn[u])continue;
			utmp=u,vtmp=v,idtmp=i;
		}else dfs1(v,u);
	}
}
inline ll max1(ll a,ll b){return a>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;
} 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值