动归——膜拜

题目描述

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

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

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

输入输出格式

输入格式：

输入文件第一行包括N 和M。

之后N 行，每行一个整数，1 表示神牛甲的崇拜者，2 表示神牛乙的崇拜者。

输出格式：

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

输入输出样例

输入样例#1：

5 1
2
2
1
2
2

输出样例#1：

2

说明

对于30%的数据，有1 ≤ N ,M≤ 50；

对于100%的数据，有1 ≤ N,M ≤ 2500。

题解：首先设变量，可把1设为1,2设为-1，检验时只要判断绝对值是否小于等于m,或有时一连串相同，即i-j==a[i]-a[j]时即可更换，注意更换时是向后遍历，因此可以把f[i]=f[j]+1;

代码：

#include<bits/stdc++.h>
using namespace std;
//f[i]表示i个人最少需要几个机房
int main()
{   int n,m,b[2501],f[2501]; 
    int a[2501];
cin>>n>>m;
for(int i=1;i<=n;i++)
{   cin>>b[i];
   if(b[i]==1)
a[i]=a[i-1]+1;
else a[i]=a[i-1]-1;
}
for(int i=1;i<=n;i++)
{ f[i]=99999999;
for(int j=0;j<i;j++)
{   if(abs(a[i]-a[j])<=m||abs(a[i]-a[j])==i-j)
   f[i]=min(f[i],f[j]+1);//向后检验 
}
}
cout<<f[n]; 

}