P1564 膜拜

题目描述

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

某学校有两位神牛，神牛甲和神牛乙。新入学的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。

 

把2看成-1，

‘然后求一段最长序列，使得这个序列的绝对值之和小于m

 

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstdlib>
using namespace std;
int n,m;
int s[4000];
int f[5000];
void read(int &n)
{
    char c='+';int x=0;bool flag=0;
    while(c<'0'||c>'9')
    {c=getchar();if(c=='-')flag=1;}
    while(c>='0'&&c<='9')
    {x=x*10+(c-48);c=getchar();}
    flag==1?n=-x:n=x;
}
int main() 
{
    read(n);read(m);
    for(int i=1;i<=n;i++)
    {
        int a;
        read(a);
        if(a==1)
            s[i]=s[i-1]+1;
        if(a==2)
            s[i]=s[i-1]-1;
    }
    for(int i=1;i<=n;i++) 
    {
        f[i]=0x7ff;
        for(int j=0;j<i;j++)
            if(abs(s[i]-s[j])<=m||abs(s[i]-s[j])==i-j) // 绝对值小于m或是同种膜拜者
                f[i]=min(f[i],f[j]+1);
    }
    cout<<f[n]<<endl;
    return 0;
}