tyvj1061移动服务——DP

Posted 2020-10-27 Zinn

题目：http://www.joyoi.cn/problem/tyvj-1061

DP记录状态为当前任务时不在此任务位置上的两个人的位置(因为一定有一个人在此任务位置上)；

不妨设初始位置p[0]=3，另外两个人在1,2位置上；

每次考虑上一任务的哪个人来到这一任务位置，更新；

开滚动数组以防止爆空间，因此注意每次i++时给f数组赋初值。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int l,n,c[205][205],p[1005],f[3][205][205],ans,INF=10000005;
int main()
{
    scanf("%d%d",&l,&n);
    for(int i=1;i<=l;i++)
        for(int j=1;j<=l;j++)
            scanf("%d",&c[i][j]);
    for(int i=1;i<=n;i++)
        scanf("%d",&p[i]);
    memset(f,2,sizeof f);
    f[0][1][2]=0;
    p[0]=3;
    for(int i=1;i<=n;i++)
    {
        memset(f[i%2],2,sizeof f[i%2]);//!!!
        for(int x=1;x<=l;x++)
            for(int y=x+1;y<=l;y++)
            {
                if(x==p[i-1]||y==p[i-1])continue;
                if(x!=p[i]&&y!=p[i])
                    f[i%2][x][y]=min(f[i%2][x][y],f[(i-1)%2][x][y]+c[p[i-1]][p[i]]);
                if(x!=p[i])
                {
                    int a=min(x,p[i-1]),b=max(x,p[i-1]);
                    f[i%2][a][b]=min(f[i%2][a][b],f[(i-1)%2][x][y]+c[y][p[i]]);
                }
                if(y!=p[i])
                {
                    int a=min(y,p[i-1]),b=max(y,p[i-1]);
                    f[i%2][a][b]=min(f[i%2][a][b],f[(i-1)%2][x][y]+c[x][p[i]]);
                }
            }
    }
    ans=INF;
    for(int x=1;x<=l;x++)
        for(int y=x+1;y<=l;y++)
            ans=min(ans,f[n%2][x][y]);
    printf("%d\n",ans);
    return 0;
}