UVALive - 4987 Evacuation Plan

最新推荐文章于 2018-08-01 10:38:07 发布

原创最新推荐文章于 2018-08-01 10:38:07 发布 · 1.2k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

动态规划

本文介绍了一种解决施工队与避难所最优分配问题的算法。通过将施工队和避难所按距离排序，并使用动态规划求解，确保每个避难所至少被一个施工队使用，同时使总移动距离最小。

题意：有n支施工队在修一条笔直的高速公路，其中第i支队伍离告诉公路起点的距离是ai,

另外还有m个避难所，其中第i个避难所离高速公路的起点的距离是bi,给每只施工队分配一个避难所，距离是|ai,-bi|,求使得所有施工队移动距离的总距离最小，每个避难所最起码有一个施工队，打印每个施工队伍的避难所号

思路：很容易想到的是dp[i][j]将前i个队伍安排进前j个避难所的最小移动距离，可以贪心的想到每只队都移动到最近的避难所，但这并不能保证所有的避难所都有队伍，所以我们先都按距离排序一下，对于第i只队伍，如果它的前一只队伍只安排到了j-1,那么i就一定要在j了，如果前i-1只安排进了j，那么由于我们已经排序好了，所以我们将i安排进j是最小的距离，为了能保证避难所都安排到，所以我们从j开始递减，还要开一个数组记录第i只在前j个避难所的位置，再接下来是由于内存的限制，我们必须用滚动数组化为一维的

#include <iostream>
#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.id < b.id;
    return a.d < b.d;
}

bool cmp1(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 j = 0; j < m; j++){
            scanf("%d",&b[j].d);
            b[j].id = j;
        }
        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--){
                if (!j || dp[j] < dp[j-1]){
                    dp[j] = 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,cmp1);
        printf("%d",a[0].ans+1);
        for (int i = 1; i < n; i++)
            printf(" %d",a[i].ans+1);
        printf("\n");
    }
    return 0;
}