P1564膜拜

题目描述

神牛有很多…当然…每个同学都有自己衷心膜拜的神牛.
某学校有两位神牛，神牛甲和神牛乙。新入学的 n 位同学们早已耳闻他们的神话。
所以，已经衷心地膜拜其中一位了。现在，老师要给他们分机房。但是，要么保证整个机房都是同一位神牛的膜拜者，或者两个神牛的膜拜者人数差不超过 m。另外，现在 n 位同学排成一排，老师只会把连续一段的同学分进一个机房。老师想知道，至少需要多少个机房。

输入格式

输入文件第一行包含两个整数 n 和 m。
第 2 到第 (n+1) 行，每行一个非 1 即 2 的整数，第 (i+1) 行的整数表示第 i 个同学崇拜的对象，1 表示甲，2 表示乙。

输出格式

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

输入输出样例

输入

5 1
2
2
1
2
2

输出

 2

数据规模与约定

对于 30% 的数据，保证 1≤n,m≤50。
对于 100% 的数据，保证 1≤n,m≤2500。

思路

所以说有很多大佬在一个学校也是很痛苦的一件事
设f[i]为第i个人为止需要的最小机房数，s[i]为从第1个到第i个人中膜拜甲、乙的人数“和”（膜拜乙的人权值为-1，甲的为1）
方程：
$若i到j的这一段人全是膜拜甲或乙的或者差<=m:dp[i]=min(dp[i],dp[j-1]+1)$
代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
int s[2550],dp[2550];
int main()
{
    memset(dp,0x7f7f7f7f,sizeof(dp));
    dp[0]=0;
 int n,m;
 cin>>n>>m;
 for (int i=1;i<=n;i++)
 {
  int x;
  cin>>x;
  if (x==1) s[i]=s[i-1]+1;
  else s[i]=s[i-1]-1;
 }
    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)
            {
                dp[i]=min(dp[i],dp[j-1]+1);
            }
        }
    }
 cout<<dp[n];
    return 0;
}