题目描述
神牛有很多...当然...每个同学都有自己衷心膜拜的神牛.
某学校有两位神牛,神牛甲和神牛乙。新入学的 n 位同学们早已耳闻他们的神话。
所以,已经衷心地膜拜其中一位了。现在,老师要给他们分机房。但是,要么保证整个机房都是同一位神牛的膜拜者,或者两个神牛的膜拜者人数差不超过 m。另外,现在 n 位同学排成一排,老师只会把连续一段的同学分进一个机房。老师想知道,至少需要多少个机房。
输入格式
输入文件第一行包含两个整数 nn 和 mm。
第 2 到第 (n + 1)行,每行一个非 1 即 2 的整数,第 (i+1) 行的整数表示第 ii 个同学崇拜的对象,1 表示甲,2 表示乙。
输出格式
输出一个整数,表示最小需要机房的数量。
输入输出样例
输入 #1复制
5 1 2 2 1 2 2
输出 #1复制
2
说明/提示
数据规模与约定
- 对于 30\%30% 的数据,保证1≤n,m≤50。
- 对于 100\%100% 的数据,保证 1≤n,m≤2500。
-
//dp[i]表示前i个人所需的最少机房数。 //前缀和+差分思想。 //如果第i个人是1,那么sum[i]=sum[i-1]+1, 否则sum[i]=sum[i-1]-1 //if(abs(sum[i]-sum[j-1]==i-j+1)),说明i到j都相同 //if(abs(sum[i]-sum[j-1]<=m)),说明i和j之间的不同人数差不超过m //在符合以上两个条件下,转移方程:dp[i]=min(dp[i],dp[j-1]+1) #include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> using namespace std; int n,m; int be[2505],sum[2505],dp[2505]; int main() { memset(dp+1,0x7f,sizeof(dp)); scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&be[i]); if(be[i]&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++)//枚举前i个 { 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); } } } printf("%d",dp[n]); return 0; }