HDU4966(确定原点的最小树形图）

题意：总共有n门课，每门也有一个最高等级ai，一开始每门都是等级0。接着有m个提升班，每个班级有a l1 b l2 w，表示只有第a门课程在l1及以上时才能上，花费代价w，可以讲第b门课程提升到l2等级。现在就将所有的课程提升到最高等级需要付出的最小代价，若无法达到则输出-1。

解题思路：裸的确定原点的最小树形图。

构图思路：所有课程的每个等级视为一个节点，对于每门课程的等级i,可以建一条对等级i-1的有向边，边权为0；对于每个提升班，可以建一条边权为w的从点(a,l1)到(b,l2)的有向边。那么根据题意，就是求这幅图的最小生成树。最小生成树的所有边的权值之和就是答案了。主要就是最后再弄个0点连向各个课程的Level 0，边权为0.

#include<bits/stdc++.h>
using namespace std;  
#define INF 0x3f3f3f3f  
#define N 550  
#define ll long long
struct edge //图的结构体  
{  
    int u,v;
	int w;  
}e[50005];  
int m,n,pre[N],id[N],visit[N],xroot,sum[N],a[N];  
int in[N],s;  
//eCnt为图中的边数  
//n为图中的顶点数  
//pre[i]为顶点i的前驱节点  
//id[i]为缩环，形成新图的中间量  
//in[i]为点i的最小入边  
//visit[i]遍历图时记录顶点是否被访问过  
int directedMST(int root,int nv,int ne)  
{  
    int ans=0;  
    while(1)  
    {  
        //1.找最小入边  
        for(int i=0;i<nv;i++) in[i]=INF;  
        for(int i=0;i<ne;i++)  
        {  
            int u=e[i].u;  
            int v=e[i].v;  
            if(u!=v&&e[i].w<in[v])  
            {   
                in[v]=e[i].w;  
                pre[v]=u;  
            }  
        }  
        for(int i=0;i<nv;i++)                          //判断图是否连通  
            if(i!=root&&in[i]==INF) return -1;  //除了跟以外有点没有入边,则根无法到达它  
         //2.找环  
        int nodeCnt=0;          //图中环的数目  
        memset(id, -1, sizeof(id));  
        memset(visit, -1, sizeof(visit));  
        in[root]=0;  
        for(int i=0;i<nv;i++)  
        {  
            ans+=in[i];  
            int v=i;  
            while(visit[v]!=i&&id[v]==-1&&v!=root)//每个点寻找其前序点，要么最终寻找至根部，要么找到一个环  
            {  
                visit[v]=i;  
                v=pre[v];  
            }  
            if(v!=root&&id[v]==-1)//缩点  
            {  
                for(int u=pre[v];u!=v;u=pre[u])  
                    id[u]=nodeCnt;  
                id[v]=nodeCnt++;  
            }  
        }  
        if(nodeCnt==0) break;//如果无环，跳出循环  
        for(int i=0; i<nv; i++)  
            if(id[i]==-1)  
                id[i]=nodeCnt++;  
        //3.缩点，重新标记  
        for(int i=0;i<ne;i++)  
        {  
            int v=e[i].v;  
            e[i].u=id[e[i].u];  
            e[i].v=id[e[i].v];  
            if(e[i].u!=e[i].v)  
                e[i].w-=in[v];  
        }  
        nv=nodeCnt;  
        root=id[root];  
    }  
    return ans;  
}  
  
int main()  
{  
	//freopen("t.txt","r",stdin);
    int n,m,c,l1,d,l2,money;  
    while(scanf("%d%d",&n,&m)!=EOF)  
    {
    	int cnt=0;
		if(n==0&&m==0) break;  
		sum[0]=1;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			a[i]++;
			sum[i]=sum[i-1]+a[i];
		}
        s=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=sum[i]-1;j>sum[i-1];j--)
			{
				e[cnt].u=j;
				e[cnt].v=j-1;
				e[cnt].w=0;
				cnt++;
			}
		}  
        for(int i=0;i<m;i++)  
        {  
        	scanf("%d%d%d%d%d",&c,&l1,&d,&l2,&money); 
        	e[cnt].u=sum[c-1]+l1;
        	e[cnt].v=sum[d-1]+l2;
        	e[cnt].w=money;
            s+=e[cnt].w;  
            cnt++;
            /*if(e[i].u==e[i].v)  
                e[i].w=INF;//消除自环  */
        }  
        s++;      //此处必须++，因为需要权值比总权值大
        for(int i=0;i<n;i++)  
        {  
            e[cnt].u=0;  
            e[cnt].v=sum[i];  
            e[cnt].w=0;  
            cnt++;
        }  
        int ans=directedMST(0,sum[n],cnt);  
        if(ans==-1 ) printf("-1\n");//ans-sum是除去虚根的最小树形图的最短路径，如果这个距离比所有的边权值和sum还大，说明还有另外的边由虚点发出，故说明此图不连通  
        else printf("%d\n", ans); 
    }  
    return 0;  
}