CF1023F Mobile Phone Network

Link
先让\(k\)条边的权值为\(0\)然后建出MST。
然后我们枚举非树边\((u,v,w)\)，树上\(u,v\)间的路径上的边的边权都必须\(\le w\)。
这个操作可以用并查集/树剖+线段树等数据结构维护。

#include<tuple>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstring>
#include<utility>
#include<iostream>
#define pi pair<int,int>
namespace IO
{
    char ibuf[(1<<21)+1],*iS,*iT;
    char Get(){return (iS==iT? (iT=(iS=ibuf)+fread(ibuf,1,(1<<21)+1,stdin),(iS==iT? EOF:*iS++)):*iS++);}
    int read(){int x=0,c=Get();while(!isdigit(c))c=Get();while(isdigit(c))x=x*10+c-48,c=Get();return x;}
}
using IO::read;
using std::fill;
using std::pair;
using std::tuple;
using std::vector;
using i64=long long;
const int N=500007;
vector<pi>e[N];vector<tuple<int,int,int>>E;
int dep[N],p[N],fa[N];
void dfs(int u,int d){dep[u]=d;for(auto[v,f]:e[u])if(!dep[v])p[v]=u*f,dfs(v,d+1);}
int find(int x){return fa[x]? fa[x]=find(fa[x]):x;}
int main()
{
    int n=read(),k=read(),m=read(),cnt=0;i64 ans=0;
    for(int i=1,u,v;i<=k;++i) u=read(),v=read(),e[u].emplace_back(v,1),e[v].emplace_back(u,1),fa[find(u)]=find(v);
    for(int i=1,u,v,fu,fv;i<=m;++i)
    if((fu=find(u=read()))^(fv=find(v=read()))) read(),e[u].emplace_back(v,-1),e[v].emplace_back(u,-1),fa[fu]=fv;
    else E.emplace_back(u,v,read());
    dfs(1,1),memset(fa+1,0,n<<2);
    for(auto[u,v,w]:E)
    while((u=find(u))^(v=find(v)))
    {
        if(dep[u]<dep[v]) std::swap(u,v);
        if(p[u]>0) ++cnt,ans+=w;
        fa[u]=abs(p[u]);
    }
    cnt<k? puts("-1"):printf("%lld",ans);
}