【C++前后缀分解动态规划】2100. 适合野炊的日子|1702

本文涉及知道点

LeetCode2100. 适合野炊的日子

你和朋友们准备去野炊。给你一个下标从 0 开始的整数数组 security ，其中 security[i] 是第 i 天的建议出行指数。日子从 0 开始编号。同时给你一个整数 time 。

如果第 i 天满足以下所有条件，我们称它为一个适合野炊的日子：

第 i 天前和后都分别至少有 time 天。

第 i 天前连续 time 天建议出行指数都是非递增的。

第 i 天后连续 time 天建议出行指数都是非递减的。

更正式的，第 i 天是一个适合野炊的日子当且仅当：security[i - time] >= security[i - time + 1] >= ... >= security[i] <= ... <= security[i + time - 1] <= security[i + time].

请你返回一个数组，包含所有适合野炊的日子（下标从 0 开始）。返回的日子可以任意顺序排列。

示例 1：

输入：security = [5,3,3,3,5,6,2], time = 2

输出：[2,3]

解释：

第 2 天，我们有 security[0] >= security[1] >= security[2] <= security[3] <= security[4] 。

第 3 天，我们有 security[1] >= security[2] >= security[3] <= security[4] <= security[5] 。

没有其他日子符合这个条件，所以日子 2 和 3 是适合野炊的日子。

示例 2：

输入：security = [1,1,1,1,1], time = 0

输出：[0,1,2,3,4]

解释：

因为 time 等于 0 ，所以每一天都是适合野炊的日子，所以返回每一天。

示例 3：

输入：security = [1,2,3,4,5,6], time = 2

输出：[]

解释：

没有任何一天的前 2 天建议出行指数是非递增的。

所以没有适合野炊的日子，返回空数组。

提示：

1 <= security.length <= 10⁵

0 <= security[i], time <= 10⁵

前后缀分解

n = security.length

left[i] 记录以nums[i]结尾的最长非递增子数组的长度。

如果left[i] <= left[i-1]则left[i] = left[i-1]+1；否则left[i] =1 。

right[i]记录以nums[i]开始的最长非递减子数组的长度。

如果left[i]和right[i]大于time，则是好日子。

right[n-1-i] 就是nums的转置数组的left[i]

代码

打开打包代码的方法兼述单元测试

核心代码

cpp 复制代码

class Solution {
		public:
			vector<int> goodDaysToRobBank(vector<int>& security, int time) {
				m_iN = security.size();
				auto Do = [&](const vector<int>& security) {
					vector<int> ret(m_iN, 1);
					for (int i = 1; i < m_iN; i++) {
						if (security[i] <= security[i - 1]) {
							ret[i] = ret[i - 1] + 1;
						}
					}
					return ret;
				};
				auto left = Do(security);
				auto right = Do(vector<int>(security.rbegin(), security.rend()));
				vector<int> ret;
				for (int i = 0; i < m_iN; i++) {
					if ((left[i] > time) && (right[m_iN - 1 - i] > time)) {
						ret.emplace_back(i);
					}
				}
				return ret;
			}
			int m_iN;
		};

单元测试

cpp 复制代码

	vector<int> security;
		int time;
		TEST_METHOD(TestMethod11)
		{
			security = { 5, 3, 3, 3, 5, 6, 2 }, time = 2;
			auto res = Solution().goodDaysToRobBank(security, time);
			AssertEx({ 2,3 }, res);
		}
		TEST_METHOD(TestMethod12)
		{
			security = { 1,1,1,1,1 }, time = 0;
			auto res = Solution().goodDaysToRobBank(security, time);
			AssertEx({ 0,1,2,3,4 }, res);
		}
		TEST_METHOD(TestMethod13)
		{
			security = { 1,2,3,4,5,6 }, time = 2;
			auto res = Solution().goodDaysToRobBank(security, time);
			AssertEx({  }, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功

视频课程

先学简单的课程，请移步CSDN学院，听白银讲师（也就是鄙人）的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了，为老板分忧，请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

测试环境

操作系统：win7 开发环境： VS2019 C++17

或者操作系统：win10 开发环境： VS2022 C++17

如无特殊说明，本算法用**C++**实现。

【C++前后缀分解 动态规划】2100. 适合野炊的日子|1702