洛谷 P2053 [SCOI2007]修车（费用流）

题目描述

同一时刻有N位车主带着他们的爱车来到了汽车维修中心。维修中心共有M位技术人员，不同的技术人员对不同的车进行维修所用的时间是不同的。现在需要安排这M位技术人员所维修的车及顺序，使得顾客平均等待的时间最小。

说明：顾客的等待时间是指从他把车送至维修中心到维修完毕所用的时间。

输入输出格式

输入格式：

 

第一行有两个数M,N，表示技术人员数与顾客数。

接下来n行，每行m个整数。第i+1行第j个数表示第j位技术人员维修第i辆车需要用的时间T。

 

输出格式：

 

最小平均等待时间，答案精确到小数点后2位。

 

输入输出样例

输入样例#1： 

2 2
3 2
1 4

输出样例#1： 

1.50

说明

(2<=M<=9,1<=N<=60), (1<=T<=1000)

解题思路

对于一个技术人员来说，假设他修了n辆车，每辆车耗时分别为w1~wn（按修理顺序），那么，等待的总时间为n*w1+(n-1)*w2+(n-2)*w3+...+2*wn-1+1*wn。所以把m个技术人员每个人分成n个工时，n辆车分别连接这m*n个工时节点，边权为w*k（第n-k+1个工时），流量为1。源点s连n量车，m*n个技术人员工时连汇点t，边权为0，流量为1。这时，流过边（x,y_z）就表示车x被安排在技术员y的第z个工时。然后求最大流的最小费用，就是最小等待时间，除以n辆车就是答案。

代码如下

#include <iostream>
#include <queue>
#include <cstring>
#include <vector>
#include <cmath>
#include <cstdio>
#define INF 0x3f3f3f3f
#define s 0
#define maxn 1005
using namespace std;
int t;
vector<int> g[maxn];
struct Line{
	int r, flow, dis;
	Line(int r, int f, int d): r(r), flow(f), dis(d){	}
};
vector<Line> line;
void add_line(int x, int y, int f, int d)
{
	//cout << x << " " << y << endl;
	line.push_back(Line(y, f, d));
	g[x].push_back(line.size() - 1);
	line.push_back(Line(x, 0, -d));
	g[y].push_back(line.size() - 1);
}
int pre[maxn];
int edge[maxn];
int dis[maxn], flow[maxn];
bool spfa()
{
	bool vis[maxn] = {0};
	memset(pre, -1, sizeof(pre));
	memset(dis, 0x7f, sizeof(dis));
	memset(flow, 0x7f, sizeof(flow));
	queue<int> que;
	que.push(s);
	vis[s] = true;
	dis[s] = 0;
	while(!que.empty()){
		int top = que.front();
		que.pop();
		vis[top] = false;
		for(int i = 0; i < g[top].size(); i ++){
			int z = g[top][i];
			int r = line[z].r;
			if(line[z].flow && dis[top] + line[z].dis < dis[r]){
				dis[r] = dis[top] + line[z].dis;
				pre[r] = top;
				edge[r] = z;
				flow[r] = min(flow[top], line[z].flow);
				if(!vis[r]){
					vis[r] = true;
					que.push(r);
				}
			}
		}
	}
	return pre[t] != -1;
}
void EK()
{
	int max_flow = 0;
	int min_dis = 0;
	while(spfa()){
		max_flow += flow[t];
		min_dis += flow[t] * dis[t];
		int now = t;
		while(pre[now] != -1){
			line[edge[now]].flow -= flow[t];
			line[edge[now]^1].flow += flow[t];
			now = pre[now];
		}
	}
	//cout << max_flow << " " << min_dis << endl;
	double ans = (double)min_dis / (double)max_flow;
	printf("%.2lf\n", ans);
}
int main()
{
	int n, m;
	while(cin >> m >> n){
		for(int i = 1; i <= n; i ++)
			add_line(s, i, 1, 0);
		t = maxn - 1;
		for(int i = 1; i <= n; i ++){
			for(int j = 1; j <= m; j ++){
				int d;
				cin >> d;
				for(int k = 1; k <= n; k ++){
					add_line(i, n + (j-1)*n + k, 1, d*k);
					if(i == 1)   //连一次就够了 
						add_line(n + (j-1)*n + k, t, 1, 0);  
				}				
			}
		}	
		EK();
		for(int i = s; i <= t; i ++)
			g[i].clear();
		line.clear(); 
	}
	return 0;
}