Codeforces Round #607 (Div. 2) E - Jeremy Bearimy(贪心+dfs)

在这里插入图片描述
在这里插入图片描述
思路:给定2k的点,以及2k-1条边构成的树,然后要k对点相连,问你相连后最小权距离和最大权距离。
思路:主要参考了大佬的代码点击链接

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int n,T,head[maxn],cnt,num[maxn];
ll maxx,minn;
struct node{
	int y,len,next;
}a[maxn<<1];
void add(int x,int y,int z)
{
	a[cnt].y=y;a[cnt].len=z;a[cnt].next=head[x];head[x]=cnt++;
}
void dfs(int u,int fa)
{
	num[u]=1;
	for(int i=head[u];i!=-1;i=a[i].next)
	{
		int v=a[i].y;
		if(v==fa) continue;
		dfs(v,u);
		if(num[v]&1) minn+=a[i].len;
		maxx+=(ll)min(n-num[v],num[v])*a[i].len;
		num[u]+=num[v];
	}
}
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		cnt=0;
		memset(head,-1,sizeof(head));
		memset(num,0,sizeof(num));
		scanf("%d",&n);
		n<<=1;
		for(int i=1;i<n;++i)
		{
			int u,v,w;
			scanf("%d %d %d",&u,&v,&w);
			add(u,v,w);add(v,u,w);
		}
		maxx=0,minn=0;
		dfs(1,0);
		printf("%lld %lld\n",minn,maxx);
	}
}
发布了39 篇原创文章 · 获赞 0 · 访问量 1061

猜你喜欢

转载自blog.csdn.net/qq_42479630/article/details/104083604