题目
你和一群强盗准备打劫银行。给你一个下标从 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;反之需要置0。递增类似,我们也可以倒序遍历。
我们将每个位置的对应的递增递减区间长度记录下来,再遍历一次判断每个位置且两边的递增递减长度大于time;即可得到每个“适合打劫银行”的日子
代码
public List<Integer> goodDaysToRobBank(int[] security, int time) {
int n = security.length;
// 非递增前缀
int[] left = new int[n];
// 非递减后缀
int[] right = new int[n];
// 一次遍历
for (int i = 1; i < n; i++) {
if (security[i] <= security[i - 1]) {
left[i] = left[i - 1] + 1;
}
// 倒序计算
if (security[n - i - 1] <= security[n - i]) {
right[n - i - 1] = right[n - i] + 1;
}
}
List<Integer> ans = new ArrayList<>();
for (int i = time; i < n - time; i++) {
// 判断左右递增递减数组
if (left[i] >= time && right[i] >= time) {
ans.add(i);
}
}
return ans;
}
时间复杂度:O(n):只需遍历数组一遍
空间复杂度:O(n):借助数组存储连续递增递减长度