洛谷 P1453 城市环路(基环树,树形dp)

传送门


解题思路

先找到基环树上的环,然后断掉任意一条环上的边,分别以两个端点做树形dp(没有上司的舞会),最后答案就是max(dp1[s][0],dp2[t][0])。
其中dp1[s][0]为以s为根节点不选s的答案,dp2[t][0]表示以t为根节点不选t时的答案。
注意断边不能只看点,因为有可能是个二元环

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<iomanip>
using namespace std;
const int maxn=1e5+5;
int p[maxn],cnt=1,a[maxn],in[maxn],s,t,n,del1,del2;
double k;
long long ans,dp[maxn][2];
struct node{
	int v,next;
}e[maxn*2];
void insert(int u,int v){
	cnt++;
	e[cnt].v=v;
	e[cnt].next=p[u];
	p[u]=cnt;
}
void topo(){
	queue<int> q;
	for(int i=1;i<=n;i++){
		if(in[i]==1) q.push(i),in[i]--;
	}
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=p[u];i!=-1;i=e[i].next){
			int v=e[i].v;
			in[v]--;
			if(in[v]==1) q.push(v);
		}
	}
}
void dfs(int u,int fa){
	dp[u][1]=a[u];
	for(int i=p[u];i!=-1;i=e[i].next){
		int v=e[i].v;
		if(v==fa||i==del1||i==del2) continue;
		dfs(v,u);
		dp[u][0]+=max(dp[v][1],dp[v][0]);
		dp[u][1]+=dp[v][0];
	}
}
int main(){
	ios::sync_with_stdio(false);
	memset(p,-1,sizeof(p));
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=n;i++){
		int u,v;
		cin>>u>>v;
		u++;v++;
		insert(u,v);
		insert(v,u);
		in[v]++;in[u]++;
	}
	cin>>k;
	topo();
	for(int u=1;u<=n;u++){
		if(s&&t) break;
		if(in[u]>=2){
			for(int i=p[u];i!=-1;i=e[i].next){
				int v=e[i].v;
				if(in[v]==2){
					s=u,t=v;
					del1=i,del2=i^1;
					break;
				}
			}
		}
	}
	dfs(s,-1);
	ans=dp[s][0];
	memset(dp,0,sizeof(dp));
	dfs(t,-1);
	ans=max(ans,dp[t][0]);
	cout<<fixed<<setprecision(1)<<k*ans;
	return 0;
}

作者:尹昱钦
欢迎任何形式的转载,但请务必注明出处。
限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。

posted @   尹昱钦  阅读(34)  评论(0编辑  收藏  举报
编辑推荐:
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· .NET 9 new features-C#13新的锁类型和语义
· Linux系统下SQL Server数据库镜像配置全流程详解
· 现代计算机视觉入门之:什么是视频
阅读排行:
· Sdcb Chats 技术博客:数据库 ID 选型的曲折之路 - 从 Guid 到自增 ID,再到
· .NET Core GC压缩(compact_phase)底层原理浅谈
· Winform-耗时操作导致界面渲染滞后
· Phi小模型开发教程:C#使用本地模型Phi视觉模型分析图像,实现图片分类、搜索等功能
· 语音处理 开源项目 EchoSharp
历史上的今天:
2020-10-25 关于float、double小项问题
点击右上角即可分享
微信分享提示