最小树形图模板 UVA11183

题意：给定n个节点m条边的有向带权图，求以0为根节点的最小树形图权值大小

用这个代码的时候要注意，这里的数据是从0开始的，边也是从0开始算，

所以在打主代码的时候，如果是从1开始，那么算法里面的从0开始的位置也要相应修改。

特别是下面的  node_circle;   

#include<cstdio>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=1e3+10;
const int maxm=4e4+10;
const int inf=0x3f3f3f3f;
int n,m;
int cost[maxn],pre[maxn],vis[maxn],id[maxn];
struct node
{
    int u,v,w;
}a[maxm];
int zhu_liu(int root)
{
    int ans=0;
    while(1){
        for(int i=0;i<n;i++) cost[i]=inf;

        for(int i=0;i<m;i++){
            int u=a[i].u,v=a[i].v,w=a[i].w;
            if(cost[v]>w&&u!=v){
                pre[v]=u;
                cost[v]=w;
            }
        }

        for(int i=0;i<n;i++){
            if(i==root) continue;
            if(cost[i]==inf) return -1;
        }

        int cnt_circle=0;
        memset(id,-1,sizeof(id));
        memset(vis,-1,sizeof(vis));
        cost[root]=0;
        for(int i=0;i<n;i++){
            ans+=cost[i];
            int p=i;
            while(vis[p]!=i&&id[p]==-1&&p!=root)
                vis[p]=i,p=pre[p];
            if(id[p]==-1&&p!=root){
                for(int u=pre[p];u!=p;u=pre[u]){
                    id[u]=cnt_circle;
                }
                id[p]=cnt_circle++;
            }
        }
        if(!cnt_circle) break;
        for(int i=0;i<n;i++)
            if(id[i]==-1) id[i]=cnt_circle++;
        for(int i=0;i<m;i++){
            int v=a[i].v;
            a[i].u=id[a[i].u];
            a[i].v=id[a[i].v];
            if(a[i].u!=a[i].v){
                a[i].w=a[i].w-cost[v];
            }
        }
        n=cnt_circle;
        root=id[root];
    }
    return ans;
}
int main()
{
    int T;
    int cnt=0;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
        int ans=zhu_liu(0);
        printf("Case #%d: ",++cnt);
        if(ans==-1) printf("Possums!\n");
        else printf("%d\n",ans);
    }
    return 0;
}

 给出从1开始的代码；

这道题是以r为根，点是从1~n,

#include<cstdio>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=1e4+10;
const ll inf=0x3f3f3f3f3f3f3f;
struct node
{
    ll u,v,w;
}a[maxn];
ll n,m;  //点的个数和边的个数都要定在全局变量，因为zhuliu里用到；
ll cost[maxn],pre[maxn],vis[maxn],id[maxn];
ll zhu_liu(ll root)
{
    ll ans=0;
    while(1){
        for(ll i=1;i<=n;i++) cost[i]=inf;

        for(ll i=1;i<=m;i++){
            ll u=a[i].u,v=a[i].v,w=a[i].w;
            if(cost[v]>w&&u!=v){
                pre[v]=u;
                cost[v]=w;
            }
        }

        for(ll i=1;i<=n;i++){
            if(i==root) continue;
            if(cost[i]==inf) return -1; //无法构成就输出-1；
        }

        ll cnt_circle=1;
        memset(id,-1,sizeof(id));
        memset(vis,-1,sizeof(vis));
        cost[root]=0;
        for(ll i=1;i<=n;i++){
            ans+=cost[i];
            ll p=i;
            while(vis[p]!=i&&id[p]==-1&&p!=root)
                vis[p]=i,p=pre[p];
            if(id[p]==-1&&p!=root){
                for(ll u=pre[p];u!=p;u=pre[u]){
                    id[u]=cnt_circle;
                }
                id[p]=cnt_circle++;
            }
        }
        if(cnt_circle==1) break;
        for(ll i=1;i<=n;i++)
            if(id[i]==-1) id[i]=cnt_circle++;
        for(ll i=1;i<=m;i++){
            ll v=a[i].v;
            a[i].u=id[a[i].u];
            a[i].v=id[a[i].v];
            if(a[i].u!=a[i].v){
                a[i].w=a[i].w-cost[v];
            }
        }
        n=cnt_circle-1;
        root=id[root];
    }
    return ans;
}
int main()
{
    ll r;
    while(scanf("%lld%lld%lld",&n,&m,&r)!=EOF){
        for(ll i=1;i<=m;i++){
            scanf("%lld%lld%lld",&a[i].u,&a[i].v,&a[i].w);
        }
        ll ans=zhu_liu(r);
        printf("%lld\n",ans);
    }
    return 0;
}