小T在玩一个游戏。游戏中,系统会给你n个城市和m条双向道路。起初,n个城市由m条道路连接,且任意一个城市都能到达另一个城市。每条道路上有一个道具,你如果摧毁了这条道路那么你就会获得这个道具并且拿到相应的分数。为了不让城市的人发现你摧毁了他们的道路,在你摧毁道路时,你需要保证任意时刻从任意一个城市出发都能到达另一个城市。现在,小T需要你编写一个程序来帮助他找出他能获得的最大分数。
输入格式:
第一行输出两个正整数n(2≤n≤2×105)和m(n−1≤m≤2×105), 表示城市数量和道路数量。
接下来m行,每行三个整数u,v,w(1≤u,v≤n,−109≤w≤109), 表示一条双向道路连接城市u和v, 摧毁这条道路获得的分数是w
注意:w不一定是正数
输出格式:
输出一行,表示能获得的最大分数。
思路: 思维+最小生成树。由于存在负边,但是负边对于答案是没有贡献的,所以直接将所有边权<=0 的边直接连接,
然后题目要求全部城市要联通,那么跑一遍Kruskal最小生成树,维持城市联通性。
那么剩下的边,就是可以获得的分数。
注意:边权最大1000 000 000 会爆int 所以得开 long long 存答案。
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e6+5;
//人间四月芳菲尽,山寺桃花始盛开。
int n,m;
struct node{
int u,v,w;
int vis,id;
}lst[N];
int s[N];
void init(int n){
for(int i=0;i<=n;i++)s[i] = i;
}
int find(int x){
if(s[x] == x)return s[x];
return s[x] = find(s[x]);
}
void merge(int x,int y){
int fx = find(x);
int fy = find(y);
if(fx != fy)s[fx] = fy;
}
bool cmp(node a,node b){
if(a.w != b.w)return a.w<b.w;
return a.id<b.id;
}
signed main(){
cin>>n>>m;
init(n);
for(int i=0;i<m;i++){
cin>>lst[i].u>>lst[i].v>>lst[i].w;
lst[i].id = i;
lst[i].vis = 0;
if(lst[i].w <=0){// 对于答案无贡献的所有边,直接连接
merge(lst[i].u,lst[i].v);//使用并查集维护城市联通性
lst[i].vis = 1;
}
}
sort(lst,lst+m,cmp);// 贪心
for(int i=0;i<m;i++){
if(lst[i].vis == 1)continue;
int u = lst[i].u;
int v = lst[i].v;
int w = lst[i].w;
if(find(u) != find(v)){//只有当当前两个城市不连通的时候,加边才是有意义的
merge(u,v);
lst[i].vis = 1;
}
}
int ans = 0;
for(int i=0;i<m;i++){
if(lst[i].vis == 1)continue;
ans += lst[i].w;
}
cout<<ans<<endl;
return 0;
}