P1564 膜拜

题目描述

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

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1

1
2
3
4
5
6
5 1
2
2
1
2
2

输出 #1

1
2

f[i] : 前 i 个同学最少需要多少个班级；

$f[i] = min(f[i],f[j]+1) ,(0<j<i)$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
const int MAX=2500+10;
int n,m,a[MAX],b[MAX],c[MAX],f[MAX];
int main(){
	freopen("in","r",stdin);
	sf(n);sf(m);
	fi(i,1,n+1) {
		sf(a[i]);
		if(a[i]==1){
			b[i] = b[i-1]+1;
			c[i] = c[i-1];
		}else {
			c[i] = c[i-1]-1;
			b[i] = b[i-1];
		} 
	}
	mem(f,0x3f);
	f[0]=0;
	fi(i,1,n+1)
		fi(j,0,i)
			if(f[j]+1<f[i]&&(abs(b[i]-b[j] + c[i]- c[j])<=m || b[i]==b[j] ||c[i]==c[j])){
				f[i]=f[j]+1;
			}
	pf(f[n]);
	return 0;
}