luogu P1564 膜拜

原题链接:

膜拜 - 洛谷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;
}