2019icpc南昌站---Bob problem

题目链接
题目大意就是有多个城市,城市之间有道路,有的道路代码为0,有的为1,代码为1的路最多可以走k条,代码为0的不限次数,求可以实现一个将所有道路连起来的路的最大值。
解题思路:首先将代码为0的路的权值都加起来,然后对代码为1的路进行贪心,贪心的方法是,首先将权值大并且是连接两个不关联的区域的,然后如果还有多的就加上权值大的。
还有就是,保留代码为1的路的信息时不要用固定的数组,要用动态数组,不然会爆。

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e4+10,M=5e5+10;
int T,n,m,k,res,f[N],vis[M],flag;
struct node{int u,v,w;}; vector<node> g;
int cmp(node a,node b){return a.w>b.w;}
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
signed main(){
	cin>>T;
	while(T--){
		cin>>n>>m>>k;	res=0;	g.clear(); vis[0]=flag=0;
		for(int i=1;i<=n;i++)	f[i]=i;
		for(int i=1,a,b,c,d;i<=m;i++){
			scanf("%lld %lld %lld %lld",&a,&b,&c,&d);
			if(!d)	res+=c,f[find(a)]=find(b);
			else	g.push_back({a,b,c}); vis[i]=0;
		}
		sort(g.begin(),g.end(),cmp);
		for(int i=0;i<g.size()&&k;i++){
			int x=find(g[i].u),y=find(g[i].v);
			if(x!=y)	k--,res+=g[i].w,f[x]=y,vis[i]=1;
		}
		for(int i=0;i<g.size()&&k;i++)	if(!vis[i])	res+=g[i].w,k--;
		for(int i=2;i<=n&&!flag;i++)	if(find(i)!=find(1))	flag++;
		printf("%lld\n",flag?-1:res); 
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值