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

分析：假设一共有n个人，那么要使所分得房间最少，那么从最后人一个开始，试图找出一个最长的可合并的序列，则找到后需要的房间数目为 这段序列需要的一间房子 和 剩余人数所需要的房子，假设其值为  1 + fs  ，则最后n个人所需房子的最小值 f[n] = min(1+fs,f[n]) ，根据此思路递推求解即可。

序列可合并的情况包括 人数差小于 m，及某个序列由同一位神牛的崇拜者组成。

代码如下：

#include <iostream>
#include <cmath>
using namespace std;
int N,M;
int f[3001];
int p[3001];
int main()  
{
	int s1 = 0,
		s2 = 0;
	for (int i = 1; i <= 3000; i++)
	{
		f[i] = 3001;
	}
	cin >> N >> M;
	for (int i = 1; i <= N; i++)
	{
		cin >> p[i];
	}
	for (int i = 1; i <= N; i++)
	{
		s1 = s2 = 0;
		int minfj = 3001;
		for (int j = i; j >= 1; j--)
		{
			if (p[j] == 1)
				s1++;
			else
				s2++;
			if (s1 == 0 || s2 == 0  ||  abs(s1 - s2) <= M)
			{
				minfj = min(minfj,f[j-1]+1);
			}
		}
		f[i] = min(f[i],minfj);
	}
	cout<<f[N];
	return 0;  
}