TJU 2248 Channel Design(最小树形图)

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=100+5;
const int maxm=10000+5;
const int inf=100000000;
int in[maxn],pre[maxn],vis[maxn],color[maxn],s[maxm],e[maxm],dis[maxm];
int n,m;
void MDST(int root)
{
    int ans=0;
    while(true)
    {
        memset(in,0x7f,sizeof(in));
        memset(pre,-1,sizeof(pre));
        for(int i=1;i<=m;i++){
            if(dis[i]<in[e[i]]&&s[i]!=e[i]){
                in[e[i]]=dis[i];
                pre[e[i]]=s[i];
            }
        }
        for(int i=1;i<=n;i++) if(i!=root&&pre[i]==-1) {printf("impossible\n");return ;}
        int cnt=0;in[root]=0;;
        memset(vis,-1,sizeof(vis)); 
        memset(color,-1,sizeof(color));
        for(int i=1;i<=n;i++){
            int u,v=i;ans+=in[i];
            while(v!=root&&vis[v]==-1) {vis[v]=i;v=pre[v];}
            if(v!=root&&vis[v]==i)
                for(color[v]=++cnt,u=pre[v];u!=v;u=pre[u]) color[u]=cnt;
        }           
        if(cnt==0) break;
        for(int i=1;i<=n;i++) if(color[i]==-1) color[i]=++cnt;
        for(int i=1;i<=m;i++){
            int u=s[i],v=e[i];
            s[i]=color[u],e[i]=color[v];
            if(color[u]!=color[v]) dis[i]-=in[v];
        }
        n=cnt;root=color[root];
    }
    printf("%d\n",ans);
}
int main()
{
    while(scanf("%d%d",&n,&m)&&(n+m))
    {
        for(int i=1;i<=m;i++) {
            scanf("%d%d%d",&s[i],&e[i],&dis[i]);
            if(s[i]==e[i]) dis[i]=inf;
        }
        MDST(1);
    }
    return 0;
}