刷题记录:牛客NC53074Forsaken喜欢独一无二的树

传送门:牛客

题目描述:

众所周知，最小生成树是指使图中所有节点连通且边权和最小时的边权子集。
不过最小生成树太简单了，我们现在来思考一个稍微复杂一点的问题。
现在给定一个n个点，m条边的图，每条边ei都有一个权值wi。定义删除一条边ei的代价为wi并且你可以对这个图执行任意次删边操作。
设这个图的最小生成树权值和为sumsum，定义一个图的最小生成树是独一无二的当且仅当这个图的边集中没有除最小生成树外的其他子集能满足权值和为sum且使得所有点连通。一个图刚开始可能没有独一无二的最小生成树，现在你可以删除一些边，使得剩下的边的最小生成树大小依然为sumsum并且这个图的最小生成树是独一无二的。
现在我们想要知道删除的边的权值和最小是多少？
输入:
1 0
输出:
0

一道最小生成树的题目.但是好像只能用$kruskal$来做??,$prim$可能做不了,如果有大佬想出来的,欢迎告诉我

前提知识:$kruskal$算法,如果不会的可以看看博主的这篇文章,对$kruskal$算法以及$prim$算法进行了较为详细的解释和证明

主要思路:

1. 首先根据我们$kruskal$的算法,我们每一次拿出来的都是两个联通分量之间最短的那条边,然后用那条边使我们的两个联通分量相连接,但是有可能我们的最短的那条边不只是单单一条的,也就是说可能有多条边的长度相同并且都是最短边,那么此时我们任选一条都是可以构成最小生成树的,此时就形成了多个最小生成树.
2. 那么我们的解法也就呼之欲出了,我们只要留下一条最短边即可.这样就保证了我们的唯一性.那么相对的,我们需要删除其他最短边.
3. 对于如何找到相同的最短边,我们发现长度显然要相等,并且两端点属于的两个连通分量和我们当前枚举到的最短边应该相同

下面是具体的代码部分:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define int long long
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
int n,m;
struct Node{
	int u,v,w;
}edge[maxn];
bool cmp(Node a,Node b) {
	return a.w<b.w;
}
int fa[maxn];
int find(int x) {
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
} 
signed main() {
	n=read();m=read();
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++) {
		edge[i].u=read();edge[i].v=read();edge[i].w=read();
	}
	sort(edge+1,edge+m+1,cmp);
	int ans=0;int cnt=0;
	for(int i=1;i<=m;i++) {
		int x=find(edge[i].u),y=find(edge[i].v);
		if(x==y) continue;
		int r=i+1;
		while(edge[r].w==edge[i].w) r++;
		for(int j=i+1;j<r;j++) {
			if((find(edge[j].u)==x&&find(edge[j].v)==y)||
			(find(edge[j].u)==y&&find(edge[j].v)==x)) {
				ans+=edge[j].w;
			}
		}
		fa[x]=y;
		cnt++;
		if(cnt==n-1) break;
	}
	cout<<ans<<endl;
	return 0;
}