CF1394D Boboniu and Jianghu

一、题目

点此看题

二、解法

我们把边定向，如果$a[u]>a[v]$，那么我们就把边定向成$v->u$

知道了这一点后考虑$dp$，我们需要解决的情况是$a[u]=a[v]$就不能直接定向，设$dp[u][0/1]$为$u$到父亲的边连入$/$连出，定向之后可以得到每个点的入度和出度，那么进行配对以后就得到了贡献是$\max (in,out)\times a_i$，考虑如何转移，似乎要枚举子树选$0$还是选$1$，时间之间爆炸。

仔细观察，发现我们只需要知道$0/1$选的数量就可以算出当前点的贡献，剩下的可以部分贪心，我们先全部强制选$0$，然后每次多选一个$1$的时候取$dp[v][1]-dp[v][0]$最小的，这样只需要排序就能解决了！

#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define int long long
const int M = 200005;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,tot,f[M],h[M],w[M],dp[M][2];//0 in 1 out
struct edge
{
	int v,next;
}e[2*M];
void dfs(int u,int p)
{
	int sum=0,a=0,b=0;vector<int> q;
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==p) continue;
		dfs(v,u);
		if(w[u]==w[v]) q.push_back(dp[v][1]-dp[v][0]),sum+=dp[v][0],a++;
		else if(w[u]>w[v]) sum+=dp[v][1],b++;
		else sum+=dp[v][0],a++;
	}
	sort(q.begin(),q.end());
	for(int i=0;i<=q.size();i++)
	{
		dp[u][0]=min(dp[u][0],sum+h[u]*max(a,b+(u!=1)));
		dp[u][1]=min(dp[u][1],sum+h[u]*max(a+(u!=1),b));
		a--;b++;
		if(i<q.size()) sum+=q[i];
	}
}
signed main()
{
	n=read();
	memset(dp,0x3f,sizeof dp);
	for(int i=1;i<=n;i++)
		h[i]=read();
	for(int i=1;i<=n;i++)
		w[i]=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,f[u]},f[u]=tot;
		e[++tot]=edge{u,f[v]},f[v]=tot;
	}
	dfs(1,0);
	printf("%lld\n",dp[1][0]);
}