UVALive 4987 Evacuation Plan

题目大意：给你n个人，m个避难点，每个人都要去一个避难点，每个避难点都不能为空，一个人 i 去一个 j 的花费为 abs( w[ i ] - w[ j ] )，当总花费最小时，让你输出最小值，并输出方案。

思路：对两个数组分别排序，设d[ i ][ j ] 表示前 i 个人，前 j 个避难点的最小花费，那么排了序，第 i 个人就肯定是去第j个避难点，当然这是前 i - 1 可以去 j 个避难点也可以是 j - 1 个，则 d[ i ][ j ] = min(d[ i -1 ][ j ] , d[ i - 1][ j - 1]) + abs(w[ i ] - w[ j ])。

第一遍自己做的时候，排了序，每次过来一个a[ i ] ，分两种情况：（1）放在第j个位置（2）二分去找和它值最接近的那个位置，现在想想没必要，因为它肯定是放在第j个位置，因为不可能让 i 前面的值来填 j 。这样dp错是没错，但是麻烦了。另外，这道题还有一个问题，那就是空间，先开始我自己敲完，一交MLE，发现空间爆了，dp数组可以滚动，但是path就没办法了，因为我要记录三个 int 值，想不出优化空间的方法了，看了别人的，回过头来，这才明白上面的道理，之后第一次敲完，path数组我存的是一个值，发现还是MLE，一个 int 四个字节啊，直到开成 bool 才过的。。 = =

代码如下：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

typedef __int64 lld;

const lld INF = 0x0fffffffffffffff ;

const int MAXN = 4004;

lld d[MAXN];

bool path[MAXN][MAXN];

int first = 1;

struct Node
{
    int num,id;
    bool operator < (const Node &tmp) const
    {
        return num<tmp.num;
    }
} a[MAXN],b[MAXN];

int ans[MAXN];

void print(int n,int m)
{
    if(n == 0 || m == 0) return ;
    print(n-1,m - path[n][m]);
    ans[a[n].id] = b[m].id;
}

int main()
{
    int n,m;
    while(~scanf("%d",&n))
    {
        for(int i = 1;i<=n;i++)
        {
            scanf("%d",&a[i].num);
            a[i].id = i;
        }
        scanf("%d",&m);
        for(int i = 1;i<=m;i++)
        {
            scanf("%d",&b[i].num);
            b[i].id = i;
        }
        sort(a+1,a+1+n);
        sort(b+1,b+1+m);
        for(int i = 0;i<=n;i++)
            d[i] = INF;
        d[1] = abs(a[1].num - b[1].num);
        for(int i = 2;i<=n;i++)
        {
            for(int j = min(m,i);j>=1;j--)
            {
                if(d[j]<=d[j-1])
                {
                    d[j] = d[j] + abs(a[i].num - b[j].num);
                    path[i][j] = 0;
                }
                else
                {
                    d[j] = d[j - 1] + abs(a[i].num - b[j].num);
                    path[i][j] = 1;
                }
                //printf("i = %d,j = %d,d[j] = %d\n",i,j,d[j]);
            }
        }
        printf("%I64d\n",d[m]);
        print(n,m);
        for(int i = 1;i<=n;i++)
            if(i==1) printf("%d",ans[i]);
            else printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}