AtCoder Beginner Contest 355（F - MST Query）

最新推荐文章于 2024-09-25 10:36:42 发布

OMG_NOIP

最新推荐文章于 2024-09-25 10:36:42 发布

阅读量1k

点赞数 13

分类专栏：题解文章标签： c++ 算法

本文链接：https://blog.csdn.net/MC_wansui/article/details/139216836

版权

题解专栏收录该内容

35 篇文章 0 订阅

订阅专栏

很久没有见到这么好的题了。

原题面

用 $C ha tGPT - 4 o$ 翻译好的题面：

你被给定一个带权无向连通图 $G$ ，它有 $N$ 个顶点和 $N - 1$ 条边，顶点编号为 $1$ 到 $N$ ，边编号为 $1$ 到 $N - 1$ 。边 $i$ 连接顶点 $a_i$ 和 $b_i$ ，权重为 $c_i$ 。

给定 $Q$ 个查询需要顺序处理。第 $i$ 个查询描述如下：

给定整数 $u_i, v_i, w_i$ 。在 $G$ 中添加一条权重为 $w_i$ 的边连接顶点 $u_i$ 和 $v_i$ 。然后，打印 $G$ 的最小生成树的边权和。

约束条件

$\leq N \leq 2 \times 10^5$
$\leq Q \leq 2 \times 10^5$
$\leq a_i < b_i \leq N$
$\leq u_i < v_i \leq N$
$\leq c_i, w_i \leq 10$

图在处理查询之前是连通的。

所有输入值均为整数。

建议还是到原题面看看样例的图。

讲解

通过数据范围发现所有边的边权都在 $10$ 以内，那么就尝试以边权为突破口。

这里有一个显著结论：

当只考虑边权为 $1$ 的边时，如果加入这条边没有形成环。那么这条边一定会一直在 $MST （最小生成树）$ 里。

接下来考虑边权只有 $1$ 和 $2$ 的情况（按照边出现的时间顺序考虑）。

如果当前的边的边权为 $1$ ，且在只考虑边权为 $1$ 的边时加入当前边不构成环，但在考虑边权有 $1$ 和 $2$ 的情况下加入当前边构成环，说明这条边最终在MST中，而且还替代掉了一个边权为 $2$ 的边。

我也知道抽象，所以来看图。

假设有三个点，三条边：

格式：端点1 $\quad$ 端点2 $\quad$ 边权

第一条： $\quad 2 \quad 1$

第二条： $\quad 3 \quad 2$

第三条： $\quad 3 \quad 1$

如果我们只考虑边权为 $1$ 的边，那么图是这样的（没有画第二条边）：

在这里插入图片描述
如果我们考虑边权为 $1$ 和 $2$ 的边，那么图是这样的：

在这里插入图片描述
在尝试插入第三条边的时候会发现如果插入这条边就会有环，所以要替代掉第二条边，在代码实现的时候在这里记录一下：有一条边权为 $2$ 的边变为了边权为 $1$ 的边，最后输出答案的时候做一下前缀和就行了。

对于其他边权的边也同理。

其中 $f l a g$ 用来判断上一轮结束的时候这条边有没有在 $MST （最小生成树）$ 中。

配合 $D S U （并查集）$ 写出代码，还有一些没有将的放到代码注释里了。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,m,a[400005],b[400005],c[400005],fas[200005],flag[400005],dp[400005];
inline ll fid(ll p){
	if(p!=fas[p])fas[p]=fid(fas[p]);
	return fas[p];
}
inline void miker(ll x,ll y){
	x=fid(x);y=fid(y);
	fas[x]=y;
}
signed main(){
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(ll i=1;i<n+m;i++)cin>>a[i]>>b[i]>>c[i];
	for(ll q=1;q<=10;q++){
		for(ll i=1;i<=n;i++)fas[i]=i;
		for(ll i=1;i<n+m;i++){
			if(c[i]>q)continue;
			if(fid(a[i])!=fid(b[i])){
				miker(a[i],b[i]);
				if(!flag[i]){
					dp[i]+=q;
					flag[i]=1;
				}
			}
			else if(flag[i]){
				// 为什么替代调的边的边权一定是 q ，因为在之前的循环中已经把 边权<q 的边能替换的都替换掉了。 
				dp[i]-=q;
				
				// 因为后面要求前缀和，所以这里要把 flag 变为 0 ，否则会重复计算。 
				flag[i]=0;
			}
		}
	}
	ll ans=0;
	for(ll i=1;i<n+m;i++){
		ans+=dp[i];
		if(i>=n)cout<<ans<<endl;
	}
	return 0;
}