最小树形图模板

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int inf=0x3f3f3f3f;
int vis[605],idx[605],pre[605],in[605],cnt;
struct node{
    int u,v;
    int w;
}edge[20005];
int solve(int t,int n,int m){//起点是t,点标号为1~n，边标号为1~m
    int x,y,i,j;
    int ans=0;
    while(1){
        for(i=1;i<=n;i++)in[i]=inf;
        for(i=1;i<=m;i++){
            node e=edge[i];
            if(e.u!=e.v&&e.w<in[e.v]){
                in[e.v]=e.w;
                pre[e.v]=e.u;
            }
        }
        for(i=1;i<=n;i++){
            if(in[i]==inf&&i!=t)return -1;
            if(i!=t)ans+=in[i];
        }
        int tot=0;
        memset(vis,-1,sizeof(vis));
        memset(idx,-1,sizeof(idx));
        for(i=1;i<=n;i++){
            y=i;
            while(vis[y]!=i&&idx[y]==-1&&y!=t){
                vis[y]=i;
                y=pre[y];
            }
            if(y!=t&&idx[y]==-1){
                tot++;
                idx[y]=tot;
                for(x=pre[y];x!=y;x=pre[x])idx[x]=tot;
            }
        }
        if(!tot)break;
        for(i=1;i<=n;i++)if(idx[i]==-1)idx[i]=++tot;
        for(i=1;i<=m;i++){
            int y=edge[i].v;
            edge[i].u=idx[edge[i].u];
            edge[i].v=idx[edge[i].v];
            if(edge[i].u!=edge[i].v)edge[i].w-=in[y];
        }
        n=tot;
        t=idx[t];
    }
    return ans;
}
int a[55],num[55][505],all;
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0)break;
        cnt=0,all=1;
        for(int i=1;i<=n;i++)   
        {
            scanf("%d",&a[i]);
            num[i][0]=all++;
            for(int j=1;j<=a[i];j++)
            {
                num[i][j]=all++;
                edge[++cnt].u=num[i][j];
                edge[cnt].v=num[i][j-1];
                edge[cnt].w=0;
            }
        }
        for(int i=1;i<=n;i++)
        {
            edge[++cnt].u=all;
            edge[cnt].v=num[i][0];
            edge[cnt].w=0;
        }
        for(int i=1;i<=m;i++)
        {
            int c,l1,d,l2,w;
            scanf("%d%d%d%d%d",&c,&l1,&d,&l2,&w);
            edge[++cnt].u=num[c][l1];
            edge[cnt].v=num[d][l2];
            edge[cnt].w=w;
        }
        int ans=solve(all,all,cnt);
        printf("%d\n",ans);
    }
    return 0;
}