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$。

/* * @Description: To iterate is human, to recurse divine. * @Autor: Recursion * @Date: 2022-08-09 23:25:49 * @LastEditTime: 2022-08-09 23:39:36 */
#include <bits/stdc++.h>
#define LL long long 
using namespace std;
const int maxn = 1e6 + 10;
const int mod = 1e9 + 7;
const int INF = 1e9 + 10;
const int N = 1e6;
int n,m;
int sum[N];
int dp[N];
int main()
{
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    fill(dp + 1,dp + 1 + n,INF); 
    for(int i = 1;i <= n;i ++){
    
        int x;
        cin >> x;
        if(x == 1)
            sum[i] = sum[i - 1] + 1;
        else
            sum[i] = sum[i - 1] - 1;
    }
    for(int i = 1;i <= n;i ++)
        for(int j = 1;j <= i;j ++)
            if(abs(sum[i] - sum[j - 1] )==i - j + 1||abs(sum[i]- sum[j - 1]) <= m)
                dp[i] = min(dp[i],dp[j - 1] + 1);
    cout << dp[n] << endl;


    return 0;
}```