题目链接
题目大意就是有多个城市,城市之间有道路,有的道路代码为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;
}