文章目录
1. 题目
2. 解题
1. 题目
你和一群强盗准备种地。给你一个下标从 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 天天气热数目是非递增的。 所以没有适合种地的日子，返回空数组。 示例 4： 输入：security = [1], time = 5 输出：[] 解释： 没有日子前面和后面有 5 天时间。 所以没有适合种地的日子，返回空数组。 提示： 1 <= security.length <= 10^5 0 <= security[i], time <= 10^5
2. 解题
- 记录连续的 >= 的次数 ct，如果 ct >= k 就满足
- 正反两次扫描，左右都满足的就是答案
class Solution { public: vector<int> goodDaysToRobBank(vector<int>& security, int time) { int n = security.size(), ct = 0; vector<int> flag(n, 0); for(int i = 0; i < n; ++i) { if(i>0 && security[i-1] >= security[i]) ct++; else ct = 0; if(ct >= time) flag[i]++; } ct = 0; for(int i = n-1; i >= 0; --i) { if(i<n-1 && security[i+1] >= security[i]) ct++; else ct = 0; if(ct >= time) flag[i]++; } vector<int> ans; for(int i = 0; i < n; ++i) if(flag[i]==2) ans.push_back(i); return ans; } };
112 ms 84.6 MB C++