最小树形图模板

朱流算法，模板里的图存储方法是结构体

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<map>
#include<cctype>
#include<stack>
#define pn putchar('
')
#define pd putchar(' ')
#define num ch-'0'
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define reint register int
#define fre1 freopen("1.txt","r",stdin);
#define fre2 freopen("2.txt","w",stdout)
using namespace std;
template <typename T>
void read(T &res)
{
    char ch;bool flag=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getchar());res=res*10+num);
    flag&&(res=-res);
}
template <typename T>
void write(T x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
typedef long long ll;
const int INF=0x7fffffff;
const int inf=0x3f3f3f3f;
const int mod=1000000007;
const int N=110;
const int M=10010;
struct node {
    int u,v,w;
}e[M];
int n,m,r;
int pre[N],id[N],f[N],minn[N];
///f[]:类似并查集，用于缩点
///id[u]:表示u在第id[u]个环中
///pre[v]:表示入度为v的最小边权对应的u
///minn[v]:表示入度为v的最小边权
int edmonds() {
    ll ans=0;
    int cnt=0;
    while(true) {
        for(int i=1;i<=n;i++)           ///重新找缩点后的最小树形图
            id[i]=f[i]=0,minn[i]=inf;
        for(int i=1;i<=m;i++)
            if(e[i].u!=e[i].v&&minn[e[i].v]>e[i].w)
            ///不是自环，且边权小于之前选定的
                minn[e[i].v]=e[i].w,pre[e[i].v]=e[i].u;
        int u=minn[r]=0;
        for(int i=1;i<=n;i++) {
            if(minn[i]==inf)
            ///即该点入度为0，没连接到，不存在最小树形图
                return -1;
            ans+=minn[i];
            for(u=i;u!=r&&f[u]!=i&&!id[u];u=pre[u]) f[u]=i;
            if(u!=r&&!id[u]) {  ///存在环,进行缩点
                id[u]=++cnt;
                for(int v=pre[u];u!=v;v=pre[v]) id[v]=cnt;
            }
        }
        if(!cnt) return ans;    ///不存在环，返回答案
        for(int i=1;i<=n;i++)
            if(!id[i]) id[i]=++cnt;  ///i结点不在树中，给它安排一个
        for(int i=1;i<=m;i++) {
            int temp=minn[e[i].v];
            if((e[i].u=id[e[i].u])^(e[i].v=id[e[i].v]))
            ///检测是否在同一个环中
                e[i].w-=temp;
                ///u=e[i].u, v=e[i].v;
                ///设u',v'为收缩后的点;
                ///当v为收缩点时,e<u,v'>=e<u,v>-minn[v];
                ///当u为收缩点时,在之前e<u,v>已经累加进入ans
                ///由每个点的入度只为1可得,e<u',v>=0;
        }
        n=cnt;r=id[r];cnt=0;
    }
}
int main()
{
    read(n),read(m),read(r);
    for(int i=1;i<=m;i++)
        read(e[i].u),read(e[i].v),read(e[i].w);
    write(edmonds());pn;
    return 0;
}