题意:
有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;
}