【codevs3369】膜拜

题目描述 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]);
}