P1564 膜拜

膜拜

题目描述

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

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

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

输入格式

输入文件第一行包含两个整数 $n$ 和 $m$。

第 $2$ 到第 $(n+1)$ 行，每行一个非 $1$ 即 $2$ 的整数，第 $(i+1)$ 行的整数表示第 $i$ 个同学崇拜的对象，$1$ 表示甲，$2$ 表示乙。

输出格式

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

样例 #1

样例输入 #1

5 1
2
2
1
2
2

样例输出 #1

2

提示

数据规模与约定

• 对于 $30\%$ 的数据，保证 $1\le n,m\le 50$。
• 对于 $100\%$ 的数据，保证 $1\le n,m\le 2500$。

#include<bits/stdc++.h>
using namespace std;
/*
f[i]:前i个同学需要的最少机房数
f[i] = min(f[i], f[j] + 1) 
将2改为-1,判断两个条件
1.区间差值正好是区间长度，则说明都是1或都是-1，必然是在同一机房
abs(s[i] - s[j - 1]) == i - j + 1
2.区间差值的绝对值<=m，这个差值的绝对值就是1比-1多的个数或者-1比1多的个数
两个条件满足其一，即可更新最小机房数。
abs(s[i] - s[j - 1]) <= m 
*/
int n, m, x, s[2505], f[2505];
int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++){
		cin >> x;
		if(x == 1) s[i] = s[i - 1] + 1;
		else s[i] = s[i - 1] - 1;
	}
	memset(f, 0x3f, sizeof f);
	f[0] = 0;
	for(int i = 1; i <= n; i ++)
		for(int j = 1; j <= i; j ++)
			if(abs(s[i] - s[j - 1]) == i - j + 1 || abs(s[i] - s[j - 1]) <= m)
				f[i] = min(f[i], f[j - 1] + 1);
	cout << f[n]; 	 

	return 0;
}