首页 > 代码库 > P1564 膜拜

P1564 膜拜

题目描述

神牛有很多…当然…每个同学都有自己衷心膜拜的神牛.

某学校有两位神牛,神牛甲和神牛乙。新入学的N 位同学们早已耳闻他们的神话。

所以,已经衷心地膜拜其中一位了。现在,老师要给他们分机房。但是,要么保证整个机房都是同一位神牛的膜拜者,或者两个神牛的膜拜者人数差不超过M。另外,现在N位同学排成一排,老师只会把连续一段的同学分进一个机房。老师想知道,至少需要多少个机房。

输入输出格式

输入格式:

 

输入文件第一行包括N 和M。

之后N 行,每行一个整数,1 表示神牛甲的崇拜者,2 表示神牛乙的崇拜者。

 

输出格式:

 

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

 

输入输出样例

输入样例#1:
5 122122
输出样例#1:
2

说明

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

对于100%的数据,有1 ≤ N,M ≤ 2500。

 

把2看成-1,

‘然后求一段最长序列,使得这个序列的绝对值之和小于m

 

 1 #include<cstdio> 2 #include<iostream> 3 #include<cmath> 4 #include<cstdlib> 5 using namespace std; 6 int n,m; 7 int s[4000]; 8 int f[5000]; 9 void read(int &n)10 {11     char c=+;int x=0;bool flag=0;12     while(c<0||c>9)13     {c=getchar();if(c==-)flag=1;}14     while(c>=0&&c<=9)15     {x=x*10+(c-48);c=getchar();}16     flag==1?n=-x:n=x;17 }18 int main() 19 {20     read(n);read(m);21     for(int i=1;i<=n;i++)22     {23         int a;24         read(a);25         if(a==1)26             s[i]=s[i-1]+1;27         if(a==2)28             s[i]=s[i-1]-1;29     }30     for(int i=1;i<=n;i++) 31     {32         f[i]=0x7ff;33         for(int j=0;j<i;j++)34             if(abs(s[i]-s[j])<=m||abs(s[i]-s[j])==i-j) // 绝对值小于m或是同种膜拜者35                 f[i]=min(f[i],f[j]+1);36     }37     cout<<f[n]<<endl;38     return 0;39 }

 

P1564 膜拜