ACM模版

描述

题解

设 dp[i][j] 表示前 i 个人疏散进掩体并且第 i 个人疏散进第 j 个掩体里的最小花费。

如此这般，状态转移方程为：

dp[i][j]=min(dp[i−1][j],dp[i−1][j−1])+abs(a[i].dis−b[j].dis)

但是这里涉及到一个问题，内存不够，会爆，所以需要用到滚动数组，不过其实不用滚动数组也可，可以优化掉第一维，首先对 a[i].dis 和 b[i].dis 进行从小到大排序，然后从后往前规划，可以取消后效性，这样就能省去表示第 i 个人这一维。

最后还要输出所有人的疏散掩体，所以在规划的过程中记录 path[i][j] 表示第 i 个人疏散进第 j 个掩体时的前一个人，也就是第 i−1 个人疏散进的掩体标号。如此，只消得一个搜索便可得出所有人的疏散目标。

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>

using namespace std;

const int MAXN = 4e3 + 10;
const long long INF = 0x3f3f3f3f3f3f3f3f;

short path[MAXN][MAXN];
long long dp[MAXN];

struct node
{
    long long dis;
    int pos, she;
};

int n, m;
node a[MAXN], b[MAXN];

bool cmp_dis(const node &a, const node &b)
{
    return a.dis < b.dis;
}

bool cmp_pos(const node &a, const node &b)
{
    return a.pos < b.pos;
}

void get_she(int i, int j)
{
    if (i != 1)
    {
        get_she(i - 1, path[i][j]);
    }
    a[i].she = b[j].pos;
}

int main(int argc, char const *argv[])
{
    int T;
    scanf("%d", &T);

    while (T--)
    {
        memset(dp, 0x3f, sizeof(dp));

        scanf("%d", &n);
        for (int i = 1; i <= n; i++)
        {
            scanf("%lld", &a[i].dis);
            a[i].pos = i;
        }

        scanf("%d", &m);
        for (int i = 1; i <= m; i++)
        {
            scanf("%lld", &b[i].dis);
            b[i].pos = i;
        }

        sort(a + 1, a + 1 + n, cmp_dis);
        sort(b + 1, b + 1 + m, cmp_dis);

        dp[1] = abs(a[1].dis - b[1].dis);
        for (int i = 2; i <= n; i++)
        {
            int t = min(i, m);
            for (int j = t; j >= 1; j--)
            {
                if (dp[j] < dp[j - 1])
                {
                    dp[j] = dp[j] + abs(a[i].dis - b[j].dis);
                    path[i][j] = j;
                }
                else
                {
                    dp[j] = dp[j - 1] + abs(a[i].dis - b[j].dis);
                    path[i][j] = j - 1;
                }
            }
        }

        printf("%lld\n", dp[m]);

        get_she(n, m);

        sort(a + 1, a + 1 + n, cmp_pos);
        for (int i = 1; i < n; i++)
        {
            printf("%d ", a[i].she);
        }
        printf("%d\n", a[n].she);
    }

    return 0;
}