首页 > 代码库 > codevs 3369 膜拜

codevs 3369 膜拜

3369 膜拜

 

 时间限制: 1 s
 空间限制: 128000 KB
 题目等级 : 黄金 Gold
 
 
题目描述 Description

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

输入描述 Input Description

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

输出描述 Output Description

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

样例输入 Sample Input

5 1 

2  


2

样例输出 Sample Output

2

数据范围及提示 Data Size & Hint

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

/*预处理一下前i位置有多少个两个神牛的崇拜者(前缀和数组),然后就是dp部分了,f[i]表示分到i个人最少多少个教室,循环i和k(i-k表示最后一个教室的人数),更新f[i]的值。预处理:f[0]=0;*/#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#define N 2501using namespace std;int dp[N],tot1[N],tot2[N],a[N];int n,m,cnt1,cnt2,ans;int main(){    scanf("%d%d",&n,&m);    for(int i=1;i<=n;i++)    {        scanf("%d",&a[i]);        if(a[i]==1) tot1[i]=1;        else tot2[i]=1;    }    for(int i=1;i<=n;i++)tot1[i]+=tot1[i-1];            for(int i=1;i<=n;i++)tot2[i]+=tot2[i-1];    memset(dp,0x3f,sizeof dp);dp[0]=0;    for(int i=1;i<=n;i++)      for(int j=0;j<i;j++)          {            if(abs((tot1[i]-tot1[j])-(tot2[i]-tot2[j]))<=m ||             tot1[i]-tot1[j]==0 ||tot2[i]-tot2[j]==0)              dp[i]=min(dp[i],dp[j]+1);        }    printf("%d\n",dp[n]);    return 0;}

 

codevs 3369 膜拜