POJ(3963)(Evacuation Plan)

链接:https://vjudge.net/problem/POJ-3963
思路:感觉状态还是很好分析出来,但是有很多细节自己根本没有考虑到导致WA声一片,第一就是输入的顺序不一定是排好序的,所以需要排序,第二是需要按照原来的输入顺序输出,所以需要开一个map记录一下,第三是因为要输出选择的避难所,不能直接更新值(最近做dp老是在犯这个错),所以需要记录一下当前状态的前一个状态是什么,然后一直dfs下去找到最初的状态然后输出。注意因为要保存前一个状态,所以最好的是由前两个状态的最优转移到当前状态而不是由前一个状态转移到两个不同的状态- -
代码:

#include
#include
#include
#include
#include
using namespace std;
int n,m;
const int maxn = 4010;  
long long dp[2][maxn];
int a[maxn],b[maxn];
map num1,num2;
bool path[maxn][maxn];
int id[maxn];

void dfs(int u,int v){
    if(u==0)return;
    id[num1[a[u]]] = num2[b[v]];
    if(path[u][v]==0)
    dfs(u-1,v-1);
    else dfs(u-1,v);
}

int main(){
    while(scanf("%d",&n)!=EOF){
        num1.clear();
        num2.clear();
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            num1[a[i]] = i;
        }
        scanf("%d",&m);
        for(int i=1;i<=m;i++){
            scanf("%d",&b[i]);
            num2[b[i]] = i;
        }
        memset(path,0,sizeof(path));
        for(int i=0;i<2;i++){
            for(int j=0;j<=m;j++){
                dp[i][j] = 1e18;
            }
        }
        sort(a+1,a+n+1);
        sort(b+1,b+1+m);
        dp[0][0]  = 0; 
        for(int i=1;i<=n;i++){
            int now = i&1;//滚动数组防止mle
            dp[now][0] = 1e18;
            for(int j=1;j<=m;j++){
                if(dp[now^1][j-1]<=dp[now^1][j]){
                dp[now][j] = dp[now^1][j-1]+abs(b[j]-a[i]); 
                path[i][j] = 0;
                }
                else{
                dp[now][j] =dp[now^1][j]+abs(b[j]-a[i]);
                path[i][j] = 1;
                }
            }
        }
        dfs(n,m);
        printf("%lld\n%d",dp[n&1][m],id[1]);
        for(int i=2;i<=n;i++)printf(" %d",id[i]);
        printf("\n");
    }    
    return 0;
}

你可能感兴趣的:(POJ(3963)(Evacuation Plan))