2016夏季练习——最小树形图

来源：互联网发布：歌曲音量增大软件编辑：程序博客网时间：2024/05/27 09:45
来源：HDU4966
终于见识到了传说中的建图问题，建图建的真是666啊。。。
代码：
#include <iostream>#include <cstdio>#include <string>#include <cstring>#include <algorithm>#include <cmath>using namespace std;const int INF = 0x3f3f3f3f;const int MAXN = 555;const int MAXE = 2555;struct Edge{    int u,v;    int w;}edge[MAXE];int level[60][510];int des[60];int pre[MAXE],id[MAXE],vis[MAXE],in[MAXE];int zhuliu(int root,int n,int m){    int ans = 0;    int v,u;    while(1){        for(int i=0;i<n;i++)            in[i] = INF;        for(int i=0;i<m;i++){            u = edge[i].u;            v = edge[i].v;            if(u!=v&&in[v]>edge[i].w){                pre[v]=u;                in[v]=edge[i].w;            }        }        for(int i=0;i<n;i++){            if(i!=root&&in[i]==INF)                return -1;        }        int tn = 0;        memset(id,-1,sizeof(id));        memset(vis,-1,sizeof(vis));        in[root] = 0;        for(int i=0;i<n;i++){            ans+=in[i];            v = i;            while(vis[v]!=i&&v!=root&&id[v]==-1){                vis[v] = i;                v = pre[v];            }            if(v!=root&&id[v]==-1){                for(u = pre[v];u!=v;u=pre[u]){                    id[u] = tn;                }                id[v] = tn++;            }        }        if(tn==0) {            return ans;        }        for(int i=0;i<n;i++){            if(id[i]==-1) id[i]=tn++;        }        for(int i=0;i<m;i++){            v = edge[i].v;            edge[i].u = id[edge[i].u];            edge[i].v = id[edge[i].v];            if(edge[i].v!=edge[i].u)                edge[i].w -= in[v];        }        n = tn;        root = id[root];    }    return ans;}int main(){    int n,m;    int ID;    while(scanf("%d%d",&n,&m)!=EOF){        if(m==0&&n==0) break;        ID=1;        for(int i=1;i<=n;i++){            scanf("%d",&des[i]);            for(int j=0;j<=des[i];j++){                level[i][j]=ID++;            }        }        int c,L1,d,L2,w;        int cnt=0;        for(int i=0;i<m;i++){            scanf("%d%d%d%d%d",&c,&L1,&d,&L2,&w);            edge[cnt].u=level[c][L1];            edge[cnt].v=level[d][L2];            edge[cnt++].w=w;        }        for(int i=1;i<=n;i++){            for(int j=1;j<=des[i];j++){                edge[cnt].v=level[i][j-1];                edge[cnt].u=level[i][j];                edge[cnt++].w=0;            }        }        for(int i=1;i<=n;i++){            edge[cnt].u = 0;            edge[cnt].v = level[i][0];            edge[cnt++].w = 0;        }        n = ID;//刚好加上0点        cout<<zhuliu(0,n,cnt)<<endl;    }return 0;}
1 0