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;
}

uvalive4987

最新文章

热门文章