移动服务(acwing 274)

题目地址

设dp(i,x,y,z)表示处理了前i个请求,且当前三个移动服务员分别在x,y,z位置的最小值。
复杂度基本为8e9,过不了。

发现可以优化一维，因为处理完第i个的时候,第i个位置一定会有服务员在。
故设dp(i,x,y) 表示三个服务员当前所在的位置分别在p[i],x,y。复杂度为4e7。

具体转移看代码注释

#include<bits/stdc++.h>
using namespace std;
const int N = 1e3 + 10,M = 210;

int dp[N][M][M],w[N][N];
int a[N];

/*
    dp[i][j][z] 表示处理到第i个，且人在个在j位置,z位置,p[i]位置的最小移动值。
    转移： 可以是第一个人移到a[i]，或者第二个人，或者第三个。(往后推比较简单)
    dp[i+1][j][z] = min(dp[i][j][z] + w[a[i]][a[i+1]],dp[i+1][j][z])
    dp[i+1][a[i]][z] = min(dp[i][j][z] + w[j][a[i+1]],dp[i+1][a[i]][z])
    dp[i+1][j][a[i]] = min(dp[i][j][z] + w[z][a[i+1]],dp[i+1][j][a[i]])
*/

int main(){
    int n,m; cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>w[i][j];
    for(int i=1;i<=m;i++) cin>>a[i];
    a[0] = 3;
    memset(dp,0x3f,sizeof(dp));
    dp[0][1][2] = 0;
    for(int i=0;i<m;i++)
        for(int j=1;j<=n;j++)
            for(int z=1;z<=n;z++){
                int p = a[i+1],t = a[i];
                if(t == j || j == z || t == z) continue;
                int v = dp[i][j][z];
                dp[i+1][t][z] = min(v + w[j][p],dp[i+1][t][z]);
                dp[i+1][j][t] = min(v + w[z][p],dp[i+1][j][t]);
                dp[i+1][j][z] = min(v + w[t][p],dp[i+1][j][z]);
            }
    int ans = 1e9;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            if(i != j && j != a[m] && i != a[m])
                ans = min(ans,dp[m][i][j]);
    cout<<ans<<endl;
    return 0;
}