BZOJ3791 作业(DP)

题意:

给出一个长度为n的01序列;

你可以进行K次操作,操作有两种:

1.将一个区间的所有1作业写对,并且将0作业写错;

2.将一个区间的所有0作业写对,并且将1作业写错;

求K次操作后最多写对了多少作业;

n<=100000,k<=50;


题解

性质:在一段区间内染色k次,每次是一段连续的,那么这个区间最多被分为2*k-1段

证明:每次染区间中间位置,最终只会染出2*k-1段(还是不明白的自己画画图

解题思路:

令dp[i][j][k]表示 当前考虑到第i本作业,1-i这段区间被染成了j段,第i本作业染为k(0或1)

则状态转移方程为

第i本作业由两种状态转移过来:

一、继承了i-1的状态,也就是原本有一次染色是染到i-1为止,现在把i也连着染一下,此时染色次数不变 即 dp[i-1][j][k]

二、从i开始新一次的染色,此时i-1的状态应该和i的状态是相反的,因为如果是相同的状态的话就没必要新一次染色了,此时染色次数要+1 即dp[i-1][j-1][k^1]

即:dp[i][j][0]=max(dp[i-1][j-1][1],dp[i-1][j][0])+(a[i]==0);
       dp[i][j][1]=max(dp[i-1][j-1][0],dp[i-1][j][1])+(a[i]==1);

 代码如下:

#include<bits/stdc++.h>
using namespace std;
int a[100010];
int dp[100010][200][2];
int n,K;
int read(){
    int ans=0;char c=getchar();
    while(!(c>='0'&&c<='9')) c=getchar();
    while(c>='0'&&c<='9'){
        ans=ans*10+c-48;
        c=getchar();
    }
    return ans;
}
int main(){
    freopen("hwk.in","r",stdin);
    freopen("hwk.out","w",stdout);
    n=read();K=read();
    for(int i=1;i<=n;i++)
        a[i]=read();
    int ans=0,i;
    if(K==0){
        printf("0");
        return 0; 
    }
    for(int j=1;j<=2*K-1;j++){
        for(i=1;i<=n;i++){
            dp[i][j][0]=max(dp[i-1][j-1][1],dp[i-1][j][0])+(a[i]==0);
            dp[i][j][1]=max(dp[i-1][j-1][0],dp[i-1][j][1])+(a[i]==1);
            ans=max(ans,max(dp[i][j][0],dp[i][j][1]));
        }
    }
    printf("%d",ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/tonyshen/p/11372329.html
今日推荐