Tokio Marine & Nichido Fire Insurance Programming Contest 2024（AtCoder Beginner Contest 355) F题

题目链接

F - MST Query

思路：

        由于是关于添加边的最小生成树的算法，我们考虑Kruskal算法。

       我们贪心的加上未加入集合的点的最短边，我们用并查集来判断是否已经处于同一个集合。

对于一条给定的a，b的边权为c的边，我们模拟Kruskal算法的过程，不断贪心的加边，直到找到

合并a，b点的时候添加的最短边，然后考虑用当前新添加的边来更新这条边。

        我们开十个并查集，对于并查集i到并查集i+1我们分别记录边权最大为i和边权最大为i+1时

合并的点。类似前缀和的思想，i+1的并查集减去i的并查集剩下的就是用Kruskal算法模拟的用

边权为i+1的边贪心添加的新边，也就是说如果i的并查集没有包括a和b，但i+1的并查集包括了a或者b，那么在生成最小生成树的时候对于a或者b点一定加了边权为i+1这条边（这条边可能只连接了a，或者只连接了b，也可能就是连接a和b）。

       在Kruskal算法中只会在连接了a点或者b点的时候考虑有关a或b的边，这时候有两种情况：1.添加的边直接连接了a和b，2.添加的边只连接了a或只连接了b。

        现在我们考虑新添的a和b边权为c的边，对于上诉的第一种情况我们直接取c和旧边的min值即可；对于第二种情况，我们设la为连接a点的边，lb为连接b点的边，我们贪心的取max（la，lb）和c比，因为取大边我们对最终最小生成树的值减少的贡献max（0，max（la，lb）-c）最大。

        所以对于并查集i+1我们考虑包括a也包括b的时候添加的边就是max（la，lb），也就是说fa[i+1][a]==fa[i+1][b]的时候的这条边。

代码：

//#include<bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
const int N=3e5+10;
const int inf=0x3f3f3f3f;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
//const ll P=2281701377;
const ll P=998244353;
#define int ll

int n,m;
int fa[15][N];
vector<int> v[N];
int find(int x,int k){
	return x==fa[k][x]?x:fa[k][x]=find(fa[k][x],k);
}
int u(int a,int b,int k){
	a=find(a,k),b=find(b,k);
	if(a==b) return 1;
	fa[k][a]=fa[k][b];
	return 0;
}
void solve(){
	cin>>n>>m;
	for(int i=1;i<=10;i++){
		for(int j=1;j<=n;j++){
			fa[i][j]=j;
		}
	}
	ll sum=0;
	for(int i=1;i<n;i++){
		int a,b,c;cin>>a>>b>>c;
		for(int j=c;j<=10;j++)
			u(a,b,j);
		sum+=c;
	}
	while(m--){
		int a,b,c;
		cin>>a>>b>>c;
		sum+=c;
		for(int j=c;j<=10;j++){
			if(u(a,b,j)){
				sum-=j;
				break;
			}
		}
		cout<<sum<<endl;
	}
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int t=1;
    //cin>>t;
    while(t--){
        solve();
    }

}