Codevs3369 膜拜

Posted 2020-09-23 Soda

3369 膜拜

 

题目描述 Description

神牛有很多…当然…每个同学都有自己衷心膜拜的神牛.
某学校有两位神牛，神牛甲和神牛乙。新入学的N位同学们早已耳闻他们的神话。所以，已经衷心地膜拜其中一位了。
现在，老师要给他们分机房。
但是，要么保证整个机房都是同一位神牛的膜拜者，或者两个神牛的膜拜者人数差不超过M。
另外，现在N位同学排成一排，老师只会把连续一段的同学分进一个机房。老师想知道，至少需要多少个机房。

输入描述 Input Description

输入文件第一行包括N和M。
之后N行，每行一个整数，1表示神牛甲的崇拜者，2表示神牛乙的崇拜者。

输出描述 Output Description

输出一个整数，表示最小需要机房的数量。

样例输入 Sample Input

5 1 
2 
2  
1 
2 
2

样例输出 Sample Output

2

数据范围及提示 Data Size & Hint

对于30%的数据，有1≤N,M≤50；
对于100%的数据，有1≤N,M≤2500

/*
  dp[i]表示前i个同学所需的最少机房数
  在一段区间j+1~i中，若这段区间内的同学满足条件，则可得到状态转移方程
      dp[i]=max{dp[j]+1}
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
using namespace std;
#define maxn 2510
int n,m,a[maxn],dp[maxn];
int main(){
    scanf("%d%d",&n,&m);
    memset(dp,127/3,sizeof(dp));
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    dp[0]=0;
    for(int i=1;i<=n;i++){
        int t1=0,t2=0;
        if(a[i]==1)t1++;
        if(a[i]==2)t2++;
        for(int j=i-1;j>=0;j--){
            if(!t1||!t2||abs(t1-t2)<=m){
                dp[i]=min(dp[i],dp[j]+1);
            }
            if(a[j]==1)t1++;
            if(a[j]==2)t2++;
        }
    }
    printf("%d",dp[n]);
}