最小树形图模板 UVA11183

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

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

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

特别是下面的  node_circle;

 #include<cstdio>
 #include<algorithm>
 #include<string.h>
 using namespace std;
 const int maxn=1e3+;
 const int maxm=4e4+;
 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=;
     while(){
         for(int i=;i<n;i++) cost[i]=inf;

         for(int i=;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=;i<n;i++){
             if(i==root) continue;
             if(cost[i]==inf) return -;
         }

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

给出从1开始的代码；

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

 #include<cstdio>
 #include<string.h>
 #include<algorithm>
 #include<queue>
 using namespace std;
 typedef long long ll;
 const int maxn=1e4+;
 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=;
     while(){
         for(ll i=;i<=n;i++) cost[i]=inf;

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

         ll cnt_circle=;
         memset(id,-,sizeof(id));
         memset(vis,-,sizeof(vis));
         cost[root]=;
         for(ll i=;i<=n;i++){
             ans+=cost[i];
             ll p=i;
             while(vis[p]!=i&&id[p]==-&&p!=root)
                 vis[p]=i,p=pre[p];
             if(id[p]==-&&p!=root){
                 for(ll u=pre[p];u!=p;u=pre[u]){
                     id[u]=cnt_circle;
                 }
                 id[p]=cnt_circle++;
             }
         }
         if(cnt_circle==) break;
         for(ll i=;i<=n;i++)
             if(id[i]==-) id[i]=cnt_circle++;
         for(ll i=;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-;
         root=id[root];
     }
     return ans;
 }
 int main()
 {
     ll r;
     while(scanf("%lld%lld%lld",&n,&m,&r)!=EOF){
         for(ll i=;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 ;
 }