3369 膜拜
题目描述 Description
神牛有很多…当然…每个同学都有自己衷心膜拜的神牛.
某学校有两位神牛,神牛甲和神牛乙。新入学的N位同学们早已耳闻他们的神话。所以,已经衷心地膜拜其中一位了。
现在,老师要给他们分机房。
但是,要么保证整个机房都是同一位神牛的膜拜者,或者两个神牛的膜拜者人数差不超过M。
另外,现在N位同学排成一排,老师只会把连续一段的同学分进一个机房。老师想知道,至少需要多少个机房。
输入描述 Input Description
输入文件第一行包括N和M。
之后N行,每行一个整数,1表示神牛甲的崇拜者,2表示神牛乙的崇拜者。
输出描述 Output Description
输出一个整数,表示最小需要机房的数量。
样例输入 Sample Input
5 1
2
2
1
2
2
样例输出 Sample Output
2
数据范围及提示 Data Size & Hint
对于30%的数据,有1≤N,M≤50;
对于100%的数据,有1≤N,M≤2500
/* dp[i]表示前i个同学所需的最少机房数 在一段区间j+1~i中,若这段区间内的同学满足条件,则可得到状态转移方程 dp[i]=max{dp[j]+1} */ #include<iostream> #include<cstdio> #include<cstring> #include<cmath> #include<cstdlib> using namespace std; #define maxn 2510 int n,m,a[maxn],dp[maxn]; int main(){ scanf("%d%d",&n,&m); memset(dp,127/3,sizeof(dp)); for(int i=1;i<=n;i++)scanf("%d",&a[i]); dp[0]=0; for(int i=1;i<=n;i++){ int t1=0,t2=0; if(a[i]==1)t1++; if(a[i]==2)t2++; for(int j=i-1;j>=0;j--){ if(!t1||!t2||abs(t1-t2)<=m){ dp[i]=min(dp[i],dp[j]+1); } if(a[j]==1)t1++; if(a[j]==2)t2++; } } printf("%d",dp[n]); }
时间: 2024-10-14 12:45:26