CF1466D 13th Labour of Heracles

知识点：度、贪心、分组

题目链接

题意

题面稍复杂。
给点权树的边染小于等于i种颜色，同颜色带两端顶点分组为一个子图，记V为图的值，v为子图的值，W为子图中连通分量的值，w为强连通分量中的点权值，定义$V=\sum v$，$v=max(W)$，$W=\sum u$，对每个i求$V_i$的最大值。

思路

易知$V_i$增加的原因是同一点权被重复计算，重复处在染色交界处（即一个节点连接两条不同颜色的边）。
可知子图只含一个强连通分量的时候最优。多个强连通分量贡献必舍其一，而舍去部分本可以由用来划分强连通分量的颜色染色所保留。
于是贪心取最大点权，由该点为分界（一个节点连接两条不同颜色的边），划分为两个子图，$V_i$增加了该点权值。
贪心，仍然取最大点权，相当于最大点权不能再作为分界为止（一个节点连接边的颜色全不同），与节点连接边数有关，联想到度（第一次尝试止步于此）。
初始状态只染一种颜色，每个节点的其中一度染上了，对最大的点权，用不同的颜色染上剩下未染色的度中，贡献一直增加最大点权。
注意，这里的颜色是抽象的，仅包含颜色“不同”的信息，不包含具体颜色的信息。

代码

#include<bits/stdc++.h>
#define ll long long
#define pii pair<ll,ll>
#define sizeof(a) sizeof(ll)*(n+5)//不写会TLE
using namespace std;
const ll N=1e5+10;

ll n;
ll arr[N];//i点权值
ll deg[N];//i点的度

void init()
{
	memset(deg,0,sizeof(deg));
}

void solve()
{
	scanf("%lld",&n);
	init();
	for(ll i=1; i<=n; i++) scanf("%lld",&arr[i]);
	ll sum=0;//求初始状态
	for(ll i=1; i<=n; i++) sum+=arr[i];
	for(ll i=1; i<n; i++)
	{
		ll u,v;
		scanf("%lld%lld",&u,&v);
		deg[u]++;
		deg[v]++;
	}
	vector<pii> sorted;//按点权降序
	for(ll i=1; i<=n; i++) sorted.push_back({arr[i],deg[i]});
	sort(sorted.begin(),sorted.end(),greater<pii>());
	printf("%lld ",sum);
	for(ll i=0; i<n; i++)//点权从大到小
	{
		for(ll j=1; j<sorted[i].second; j++)//度-1次，初始状态已经使用了一个度
			printf("%lld ",sum+=sorted[i].first);//增加贡献同时输出
	}
	puts("");
}

int main()
{
	ll t;
	scanf("%lld",&t);
	while(t--)
		solve();
	return 0;
}