LA 4987 背包

#include <bits/stdc++.h>

using namespace std;

const int maxn = 4000+5;
const int inf = 0x3f3f3f3f;

struct node {
    int d;
    int id;
    int ans;
}A[maxn],B[maxn];

long long dp[maxn];
int path[maxn][maxn];
int n,m;

bool cmp(node a,node b) {
    if(a.d==b.d)
        return a.id < b.id;
    return a.d < b.d;
}

void find_path(int i,int j) {
    if(i)
        find_path(i-1,path[i][j]);
    A[i].ans = B[j].id;
}

int cmp1(node a,node b) {
    return a.id < b.id;
}

int main()
{
    while(~scanf("%d",&n)) {
        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,inf,sizeof(dp));
        dp[0] = abs(A[0].d-B[0].d);

        for(int i=1;i<n;i++) {
            for(int j=min(m-1,i);j>=0;j--) {
                if(!j||dp[j]<dp[j-1]) {
                    path[i][j] = j;
                    dp[j] = dp[j] + abs(A[i].d-B[j].d);
                }
                else {
                    path[i][j] = j-1;
                    dp[j] = dp[j-1] + abs(A[i].d - B[j].d);
                }
            }
        }

        printf("%lld\n",dp[m-1]);
        find_path(n-1,m-1);
        sort(A,A+n,cmp1);

        for(int i=0;i<n;i++)
            printf("%d ",A[i].ans+1);
        puts("");


    }
    return 0;
}