“蔚来杯“2022牛客暑期多校训练营10-K You are given a tree...

ren1xin

已于 2022-08-21 14:35:34 修改

阅读量233

点赞数

分类专栏： “蔚来杯“ 多校联赛文章标签：树结构动态规划随机化优化生成边权值和

于 2022-08-21 14:26:55 首次发布

本文链接：https://blog.csdn.net/qq_60252920/article/details/126450230

版权

多校联赛同时被 2 个专栏收录

40 篇文章 0 订阅

订阅专栏

“蔚来杯“

9 篇文章 0 订阅

订阅专栏

原题题面：https://ac.nowcoder.com/acm/contest/33195/K

题目大意

给定包含 $n(1\le n\le 5000)$ 个点的树 $T = (V, E)$ ，节点 $i$ 具有权值 $a_i(1\le a_i\le n)$ ，每条边具有一定边权值 $w_i(-10^9\le w_i\le10^9)$ 。

求子集 $S\subseteq V$ ，满足最多有 $k(2\le k\le5)$ 个点具有不同的权值，且最大化“生成边”的权值和。边 $e$ 是生成边，当且仅当存在 $u,v\in S$ ，边 $e$ 在 $u, v$ 的简单路径上。

输出最大的“生成边”的权值和。

解题思路

考虑若点权值种数较少，可以采用树状数组和状压 $d p$ ，设 $dp_{x,V}$ 表示 $x$ 节点的子树中，已选点权种类二进制为 $V$ 的最大边权值，枚举儿子 $x$ ，枚举 $x$ 的方案 $S$ ，则其他儿子的方案为 $V - S$ ，进行 $d p$ 转移即可，设点全集为 $A$ 。

由此可得转移式：
$dp_{x,V}=max(dp_{x,V},dp_{x,V-S}+dp_{v,S}+w_s)(V\subseteq A,S\subseteq V,s\in son_x)$

但本题权值最多有 $n$ 种，需枚举 $x, v, S, V$ ，总复杂度为 $O(N3^n)$ ，显然不行，可以用随机化优化，将每个点的权值处理为 $[1, k]$ ，从而降低复杂度，处理200次，可将正确率升至 $99.96\%$ ，时间复杂度 $O(TN3^k)$ ，最多约大于 $2\times10^8$ ，勉强可过。

代码实现

#include<bits/stdc++.h>
#define ll long long
using namespace std;
template <class T> inline void read(T&x){
	char c,f=' ';
	while(!isdigit(c=getchar()))f=c;
	x=c^48;
	while(isdigit(c=getchar()))x=(x<<3)+(x<<1)+(c^48);
	if(f=='-')x=-x;
}
struct node{
	int v;
	ll w;
	node(int v,int w):v(v),w(w){}
};
const int N=5e3+5,T=300;
const ll inf=0x3f3f3f3f3f3f3f;
vector<node>ve[N];
int n,k,a[N],b[N],c[N];
ll ans,f[N][1<<5],t;
void dfs(int u,int fa){
	for (int S=0;S<1<<k;++S)f[u][S]=-inf;
    f[u][0]=0;
    f[u][1<<b[u]]=0;
	for(int i=0;i<ve[u].size();i++){
		int v=ve[u][i].v;
		if(v==fa)continue;
		dfs(v,u);
        for(int V=0;V<1<<k;V++)f[v][V]+=ve[u][i].w;
		for(int V=(1<<k)-1;V;V--){
			for(int A=V;A;A=(A-1)&V){
				f[u][V]=max(f[u][V],f[u][V-A]+f[v][A]);
				if(V-A)ans=max(ans,1ll*f[u][V-A]+f[v][A]);
			}
		}
	}
}
int main(){
	read(n),read(k);
	for(int i=1;i<=n;i++)read(a[i]);
	for(int i=1;i<n;i++){
		int u,v;ll w;
		read(u),read(v),read(w);
		ve[u].push_back(node(v,w));
		ve[v].push_back(node(u,w));
	}
	for(t=1;t<=T;t++){
		for(int i=1;i<=n;i++)c[i]=rand()%k;
        for(int i=1;i<=n;i++)b[i]=c[a[i]];
		dfs(1,0);
	}
	cout<<ans;
}