题意:输入n门课,m条路,然后n个整数,表示这n门课要修到的最高等级,m行路每行5个数 a d1 b d2 x;表示的意思是从要花x元钱直接升级到 b门课的d2等级的前提 是你的a课 要到打d1等级。 要你求 将所有课程等级过完的最小花费。
思路:将每门课的每个等级看成一个点,每个点都在图中,每个点只有一个父亲,因为他只能有其他一门课 的前提,所以这是颗,也就是求最小生成树,但是这是有向的边。所以是最小树形图。
做法 就是 先将课程的高的等级向低的等级连一条有向边花费为0,因为一门课等级高的过了,那么低等级也是过了。
然后根据他输入的边建立相应的有向边,最后将每个课程的0等级的那个点 超点汇聚一个点,那个点就是最小树形图的跟了;
代码:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
using namespace std;
#define MAX 55555
#define INF 999999999
struct edge
{
int from,to,w;
};
vector<edge>edges;
int n,m;
int sum[MAX];
int kemu[MAX];
int pre[MAX];
int in[MAX];
int vis[MAX];
int id[MAX];
void addedge(int x,int y,int w)
{
edge v={x,y,w};
edges.push_back(v);
}
int Directed_MST(int root, int V)
{
int ret = 0;
while(true)
{
//1.找最小入边
for(int i = 1; i <=V; i++) in[i] = INF;
for(int i = 0; i < edges.size(); i++)
{
int u = edges[i].from;
int v = edges[i].to;
if(edges[i].w < in[v] && u != v) {pre[v] = u; in[v] = edges[i].w;}
}
/* for(int i=1;i<=V;i++)
{
printf("%d ",pre[i]);
}
printf("\n");
for(int i=1;i<=V;i++)
printf("%d ",in[i]);
printf("\n");*/
for(int i = 1; i <=V; i++)
{
if(i == root) continue;
if(in[i] == INF) return -1;//除了跟以外有点没有入边,则根无法到达它
}
//2.找环
int cnt = 1;
memset(id, -1, sizeof(id));
memset(vis, -1, sizeof(vis));
in[root] = 0;
for(int i = 1; i <=V; i++) //标记每个环
{
ret += in[i];
int v = i;
while(vis[v] != i && id[v] == -1 && v != root) //每个点寻找其前序点,要么最终寻找至根部,要么找到一个环
{
vis[v] = i;
v = pre[v];
}
if(v != root && id[v] == -1)//缩点
{
for(int u = pre[v]; u != v; u = pre[u]) id[u] = cnt;
id[v] = cnt++;
}
}
if(cnt == 1) break; //无环 则break
for(int i = 1; i <= V; i++)
if(id[i] == -1) id[i] = cnt++;
//3.建立新图
for(int i = 0; i < edges.size(); i++)
{
int u = edges[i].from;
int v = edges[i].to;
edges[i].from = id[u];
edges[i].to = id[v];
if(id[u] != id[v]) edges[i].w -= in[v];
}
V = cnt-1;
root = id[root];
// printf("%d %drd\n",root,cnt);
}
return ret;
}
void init()
{
edges.clear();
for(int i=1;i<=n;i++)
{
scanf("%d",&kemu[i]);
kemu[i]++;//加上等级为0的点
if(i==1) sum[i]=kemu[i];
else sum[i]=sum[i-1]+kemu[i];//累加相邻两点的和 方面建立图
}
/* for(int i=1;i<=n;i++)
{
printf("%d %d\n",sum[i],kemu[i]);
}*/
for(int i=1;i<=n;i++)
{
for(int j=sum[i];j>sum[i]-kemu[i];j--)
{
for(int k=j-1;k>sum[i]-kemu[i];k--)
{
addedge(j,k,0);
}
}
}
for(int i=1;i<=m;i++)
{
int a,d1,b,d2,m;
scanf("%d %d %d %d %d",&a,&d1,&b,&d2,&m);
addedge(sum[a]-kemu[a]+1+d1,sum[b]-kemu[b]+1+d2,m);
}
for(int i=1;i<=n;i++)
{
addedge(sum[n]+1,sum[i]-kemu[i]+1,0);
}
}
int main()
{
while(scanf("%d%d",&n,&m)&&n&&m)
{
init();
/* for(int i=0;i<edges.size();i++)
{
printf("%d %d %d\n",edges[i].from,edges[i].to,edges[i].w);
}*/
// printf("%d\n",edges.size());
printf("%d\n", Directed_MST(sum[n]+1,sum[n]+1));
}
}