uvalive4987

题意:

有m个避难所n个修工队,给出各自距离公路的最短距离,要求所有的避难所都要有队并且要求距离最短

思路:

对避难所的距离以及修工队的距离进行排序,那么如果要确保所有的避难所都要住人首先要一一对应,在对应完后在考虑重住的情况

那么这一操作带来的结果就是在考虑完对应避难所后,递减来的前一个是重住到前一个的是最优解,再前一个是一个重住到在前一个的最优,实际上每递减一次重住的数量级就会增加一个(毕竟一直到重住到第一个营,也就是枚举的部队都住在第一个)

如此类推输出来的dp[m-1]就是最优解

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 4010;
struct node {
	int d;
	int id;
	int ans;
}a[maxn], b[maxn];
bool cmp(node a, node b) {
	if(a.d != b.d) return a.d<b.d;
	else return a.id<b.id;
}
bool cmp2(node a, node b) {
	return a.id<b.id;
}
long long dp[maxn];
int path[maxn][maxn];
int n, m;
void print(int i, int j) {
	if(i) print(i-1, path[i][j]);
	a[i].ans = b[j].id;
}


int main() {
	while(scanf("%d",&n) != EOF) {
		for(int i=0; i<n; i++) {
			scanf("%d", &a[i].d);
			a[i].id = i;
		}
		scanf("%d", &m);
		for(int i=0; i<m; i++) {
			scanf("%d", &b[i].d);
			b[i].id = i;
		}
		sort(a, a+n, cmp);
		sort(b, b+m, cmp);
		memset(dp, 0x3f3f3f3f, sizeof(dp));
		dp[0] = abs(a[0].d-b[0].d);
		for(int i=1; i<n; i++) {
			for(int j=min(i,m-1); j>=0; j--) {实际上每次递减所有的dp都含有一样数目的避难所了,这样可以比较出
				if(!j || dp[j]<dp[j-1]) {//重住前一个非优解
					dp[j] += abs(a[i].d - b[j].d);
					path[i][j] = j;
				}
				else {
					dp[j] = dp[j-1]+abs(a[i].d-b[j].d);//采取前面的方案
					path[i][j] = j-1;
					}
			}
		}
		printf("%lld\n", dp[m-1]);
		print(n-1, m-1);
		sort(a, a+n, cmp2);
		printf("%d", a[0].ans+1);
		for(int i=1; i<n; i++) printf(" %d", a[i].ans+1);
		printf("\n");
	}
	return 0;
}