Range Connect MST

Range Connect MST

之前已经遇到过一次类似这种问题，首先是线段树优化，其次就是并查集优化。

类似第一次修改区间后续操作不选这个区间，那么可以采用并查集的方法跳过当前区间。

那么这道题目首先将c从小到大排序，然后选区间，如果这段区间内有点已经被赋值，那么就跳过这个点，用并查集去实现。

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 2e5+5;
int n,q;
int pre[N];
struct node{
    int l,r,c;
}a[N];

int cmp(node a,node b){
    return a.c<b.c;
}

int root(int x){
    return pre[x] = (pre[x]==x)?x:root(pre[x]);
}

void solve(){
    cin>>n>>q;
    for(int i=1;i<=q;i++){
        cin>>a[i].l>>a[i].r>>a[i].c;
    }
    for(int i=1;i<=n;i++)pre[i] = i;
    sort(a+1,a+1+q,cmp);

    int ans = 0;
    for(int i=1;i<=q;i++){
        int w = a[i].c;
        int l = a[i].l;
        int r = a[i].r;
        ans += w;
        while(root(l)!=root(r)){
            pre[root(l)] = root(l) + 1;
            ans += w;
        }
    }

    if(root(1)!=root(n))cout<<-1<<"\n";
    else cout<<ans<<"\n";


}

signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int T=1;
    while(T--){
        solve();
    }

    return 0;
}