Codeforces 1637F 思维+树形dp

stdforces

于 2022-10-26 16:29:45 发布

阅读量191

点赞数

文章标签：深度优先算法

本文链接：https://blog.csdn.net/stdforces/article/details/127535212

版权

题意：

给你一棵树，每个点有一个权值 $val_{i}$ ，你需要在树上的某些点放置权值 $b_{i}$ ，不在某个点放置就意味这个点的权值 $b_{i}=0$ ，最后使得对于任意结点 $x$ ，均存在有两个 $u, v$ ，使得路径 $(u, v)$ 经过 $x$ ，且 $min(b_{u},b_{v})\geq val_{x}$ ，求出最小放置的总权值

Solution：

对于这种多限制的问题，似乎将限制最苛刻的点放置为根是一个优秀的解法（这题和2021河南省赛的MEXTREE很类似）。

将 $v a l$ 最大的点设为根，那么对于根，只需要至少一棵子树内出现过 $b$ 权值至少为 $val_{rt}$ 的就可以了，因为自己也可以是 $val_{rt}$ 的

那么对于其他任意子树 $u$ ，只需要子树内出现过一个 $b$ 比 $v a l [u]$ 大的即可，叶子的 $b$ 显然是自己的 $v a l$ ，那么我们在递归返回的时候来调整子树的 $b$ ，如果子树的权值都没有达到 $val_{u}$ 的，我们只需要对某个叶子结点加上一部分，为了最小化，对子树的最大值加上这个差值是最好的，所以需要记录目前的子树最大值。

对于根结点，则需要两个点来支撑他，所以当根只有一颗子树的时候，我们只能对这棵子树的一个叶子和根本身更新了，否则我们能在两颗子树的叶子上分别更新，此时最好的策略是更新两个 $b$ 最大的叶子，此时只需要把根的所有子树最大值排序即可

// #include<bits/stdc++.h>
#include<iostream>
#include<queue>
#include<cstdio>
#include<bitset>
#include<cstring>
#include<algorithm>
using namespace std;

using ll=long long;
const int N=200005,inf=0x3fffffff;
const long long INF=0x3f3f3f3f3f3f,mod=998244353;

struct way
{
	int to,next;
}edge[N<<1];
int cnt,head[N];

void add(int u,int v)
{
	edge[++cnt].to=v;
	edge[cnt].next=head[u];
	head[u]=cnt;
}

ll ans;
int n,val[N],max1[N],sum[N],rt;

void dfs(int u,int fa)
{
	bool flag=false;
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa) continue;
		dfs(v,u);
		flag=true;
		max1[u]=max(max1[u],max1[v]);
	}
	if(!flag) max1[u]=val[u],ans+=val[u];
	else
	{
		if(u==rt)
		{
			vector<int>tmp;
			for(int i=head[u];i;i=edge[i].next) tmp.push_back(max1[edge[i].to]);
			sort(tmp.begin(),tmp.end(),[&](const int x,const int y){
				return x>y;
			});
			if(tmp.size()==1) ans+=(val[u]>max1[u])*(val[u]-max1[u])+val[u];
			else ans+=val[u]-tmp[0]+val[u]-tmp[1];
		}
		else if(max1[u]<val[u]) ans+=val[u]-max1[u],max1[u]=val[u];
	}
}

int main()
{
	/**
	 * 给你一棵树，每个点都有一个值val[i]。你要在一些点上放值b[i]，使得对于任意一个点x，都有两个点u,v：x在u,v的路径上且min(b[u],b[v])>=a[x]
	 * 问你最少的b之和是多少。
	*/
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>val[i];
		if(val[i]>val[rt]) rt=i;
	}
	for(int i=1;i<n;i++)
	{
		int u,v; cin>>u>>v;
		add(u,v); add(v,u);
	}
	dfs(rt,0);
	cout<<ans;
	return 0;
}