Codeforces 1257 E - The Contest dp写法

题意

题目传送门
系统随机分配给三个人共n个问题，每个人可以把他所分配到的问题转移一个给其他人。问最少转移多少次使其满足第一个人剩下的问题是前缀，第三个人剩下的问题是后缀。可以存在有人没有问题。

思路

首先可以理解到，最后分配的情况肯定是第1个人解决前k个任务，第2个人解决k+1到k2的任务，第3个人解决k2+1到n的任务。(0<=k<=k2<=n),既对问题i,j来说，若i<j,则i对应解决的人编号<=j对应解决的人编号，所以不妨对n个任务所交给的人进行dp维护。dp[i][k]代表把第i个问题给予编号k的人所对应的最小转移次数

for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){//i-1问题给j人
                for(int k=j;k<=3;k++){//对应前问题给j来说保证i问题给k使k>=j
                    dp[i][k]=min(dp[i][k],dp[i-1][j]+(a[i]!=k));//a[i]代表第i个问题起始分配到的人编号
                }
            }
        }

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll maxn=2e5+5;
ll a[maxn],dp[maxn][4];
int main(){
    ll n,n1,n2,n3,x;
    while(scanf("%lld%lld%lld",&n1,&n2,&n3)==3){
        for(int i=1;i<=n1;i++) {
                scanf("%lld",&x);
               a[x]=1;
        }
        for(int i=1;i<=n2;i++){
                scanf("%lld",&x);
                a[x]=2;
        }
        for(int i=1;i<=n3;i++){
                scanf("%lld",&x);
               a[x]=3;
        }
        n=n1+n2+n3;
        	memset(dp, 63, sizeof dp);
        dp[0][1]=0;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=3;j++){
                for(int k=j;k<=3;k++){
                    dp[i][k]=min(dp[i][k],dp[i-1][j]+(a[i]!=k));
                }
            }
        }
        cout<<min(dp[n][1],min(dp[n][2],dp[n][3]))<<endl;
    }
}