60分求救
查看原帖
60分求救
151647
sycqwq楼主2021/10/10 20:52
#include<bits/stdc++.h>
#define int long long
#define INF 1926081700000000
using namespace std;
const int maxn=3e5+5;
int n,m;
typedef long long ll;
struct node
{
    int u,v,w;
}g[maxn];
int fa[maxn/3],bk[maxn],f[maxn/3][25],ma[maxn][25],cm[maxn][25],d[maxn];
vector<int> e[maxn/3],w[maxn/3];
int cmp(node s1,node s2)
{
    return s1.w<s2.w;
}
int getfa(int x)
{
    if(fa[x]==x)
    {
        return x;
    }
    return fa[x]=getfa(fa[x]);
}
void dfs(int x,int fa)
{
    d[x]=d[fa]+1ll;
    f[x][0]=fa;
    for(int i=0;i<e[x].size();i++)
    {
        int v=e[x][i];
        if(v!=fa)
        {
            ma[v][0]=w[x][i];
            cm[v][0]=-INF;
            dfs(v,x);
        }
    }
}
int matree;
int lca(int x,int y,int w)
{
    if(d[x]<d[y])
    {
        swap(d[x],d[y]);
    }
    int s=0;
    for(int i=20;i>=0;i--)
    {
        if(d[f[x][i]]>d[y])
        {
            if(ma[x][i]!=w)
            {
                s=max(s,ma[x][i]);
            }
            else
                s=max(s,cm[x][i]);
            x=f[x][i];
        }
    }
    for(int i=20;i>=0;i--)
    {
        if(f[x][i]!=f[y][i])
        {
            if(ma[x][i]!=w)
            {
                s=max(s,ma[x][i]);
            }
            else
                s=max(s,cm[x][i]);
            x=f[x][i];
            y=f[y][i];
        }
    }
    if(ma[x][0]!=w)
    {
        s=max(s,ma[x][0]);
    }
    else
        s=max(s,cm[x][0]);
    if(ma[y][0]!=w)
        s=max(s,ma[y][0]);
    else
        s=max(s,cm[y][0]);
    return s;
}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
        cin>>g[i].u>>g[i].v>>g[i].w;
    sort(g+1,g+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int x=getfa(g[i].u),y=getfa(g[i].v);
        if(x!=y)
        {
            fa[x]=fa[y];
            matree+=g[i].w;
            e[g[i].u].push_back(g[i].v);
            e[g[i].v].push_back(g[i].u);
            w[g[i].u].push_back(g[i].w);
            w[g[i].v].push_back(g[i].w);
            bk[i]=1;
        }
    }
    dfs(1,0);
    for(int i=1;i<=20;i++)
    {
        for(int j=1;j<=n;j++)
        {
            f[j][i]=f[f[j][i-1]][i-1];
            ma[j][i]=max(ma[j][i-1],ma[f[j][i-1]][i-1]);
            cm[j][i]=max(cm[j][i-1],cm[f[j][i-1]][i-1]);
            if(ma[j][i-1]>ma[f[j][i-1]][i-1])
            {
                cm[j][i]=max(cm[j][i],ma[f[j][i-1]][i-1]);
            }
            else if(ma[j][i-1]<ma[f[j][i-1]][i-1])
                cm[j][i]=max(cm[j][i],ma[j][i-1]);
        }
    }
    int ans=INT_MAX;
    for(int i=1;i<=m;i++)
    {
        if(!bk[i])
        {
            ans=min(ans,matree-lca(g[i].u,g[i].v,g[i].w)+g[i].w);
        }
    }
    cout<<ans;
    return 0;
}
2021/10/10 20:52
加载中...