题目描述 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
f[i]表示前i个人分的最小的机房数
刚开始题目没有读懂,注意一下“连续一段分到一个机房的意思”
【代码】
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
struct hp{
int a,b;
}s[2505];
int n,m,x,p,q,r;
int f[2505];
int main(){
scanf("%d%d",&n,&m);
for (int i=1;i<=n;++i){
scanf("%d",&x);
s[i].a=s[i-1].a;
s[i].b=s[i-1].b;
if (x==1) s[i].a++;
if (x==2) s[i].b++;
}
f[1]=1;
for (int i=2;i<=n;++i){
f[i]=f[i-1]+1;
for (int j=1;j<i;++j){
p=s[i].a-s[j-1].a; q=s[i].b-s[j-1].b;
r=abs(p-q);
if (r<=m||p==0||q==0)
f[i]=min(f[i],f[j-1]+1);
}
}
printf("%d",f[n]);
}