原题链接:
膜拜 - 洛谷https://www.luogu.com.cn/problem/P1564
题目描述
神牛有很多...当然...每个同学都有自己衷心膜拜的神牛.
某学校有两位神牛,神牛甲和神牛乙。新入学的 nn 位同学们早已耳闻他们的神话。
所以,已经衷心地膜拜其中一位了。现在,老师要给他们分机房。但是,要么保证整个机房都是同一位神牛的膜拜者,或者两个神牛的膜拜者人数差不超过 mm。另外,现在 nn 位同学排成一排,老师只会把连续一段的同学分进一个机房。老师想知道,至少需要多少个机房。
输入格式
输入文件第一行包含两个整数 n 和 m。
第 2 到第 (n + 1) 行,每行一个非 1 即 2 的整数,第 (i+1) 行的整数表示第 i 个同学崇拜的对象,1 表示甲,2 表示乙。
输出格式
输出一个整数,表示最小需要机房的数量。
输入输出样例
输入 #1复制
5 1 2 2 1 2 2
输出 #1复制
2
说明/提示
数据规模与约定
- 对于 100% 的数据,保证 1≤n,m≤2500。
Solution:
一题线性动态规划,把神牛2改成神牛-1然后使用前缀和统计起来。这样可以保证给定一个随意的区间端点保证可以知道里面神牛1和神牛2数量的差值是多少,如果差值小于m或者差值等于区间长度便是合法区间
接下来便是基本的动态规划了,定义dp[i]为前i个人需要的最少机房数然后枚举小于i的j,判断i j能不能作为一段,如果可以,那么尝试转移
dp[i] = min(dp[i],dp[j-1]+1)
code:
#include<bits/stdc++.h>
using namespace std;
int dp[2555];
int sum[2555];
int n,m;
int main()
{
cin>>n>>m;
for(int i = 1;i<=n;i++)
{
cin>>sum[i];
if(sum[i]==1) sum[i] = sum[i-1]+1;
else sum[i] = sum[i-1]-1;
}
memset(dp,0x3f3f3f3f,sizeof(dp));
dp[0] = 0;
for(int i = 1;i<=n;++i)
for(int j = 1;j<=i;++j)
if(abs(sum[i]-sum[j-1])<=m||abs(sum[i]-sum[j-1])==i-j+1)
dp[i] = min(dp[i],dp[j-1]+1);
cout<<dp[n]<<endl;
return 0;
}