2100.适合打劫银行的日子
你和一群强盗准备打劫银行。给你一个从下标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+1]<=......<=security[i+time-1]<=security[i+time-1]。
请你返回一个数组，包含所有适合打劫银行的日子(下标从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<=
- 0<=security[i], time<=
解法1：暴力求解法(双循环-数据量大导致超时)
1.遍历security数组中下标从time开始直到len(security)-1-time的元素；
2.再用一个循环判断当前元素的前time个元素是否是非递增的以及后time个元素是否是非递减的；
3.若前后time个元素均满足条件则将该元素的下标加入结果中。
func goodDaysToRobBank(security []int, time int) (ans []int) {
if time*2 + 1 > len(security) {
ans = nil
}
if time == 0 {
for i:=0; i<len(security); i++ {
ans = append(ans, i)
}
}
for i:=time; i<len(security)-time; i++ {
flag := false
//判断是否是适合打劫的日子
for j:=0; j<time; j++ {
if security[i-j-1] >= security[i-j] && security[i+j+1] >= security[i+j] {
flag = true
} else {
flag = false
break
}
}
if flag == true {
ans = append(ans, i)
}
}
return
}
解法2：动态规划(两次单循环)
判断第i天是否为适合打劫银行的日子，只需判断第i天之前是否有至少连续的time天警卫数量不递增并且之后是否也有连续的time天警卫数量不递减。
1.使用数组left中元素left[i]记录第i天之前连续的警卫数量不递增的天数；
2.使用数组right中元素right[i]记录第i天之后连续的警卫数量不递减的天数；
3.遍历日子并判断第i天是否满足条件(前time天非递增后time天非递减即left[i]>=time&&right[i]>=time)
func goodDaysToRobBank(security []int, time int) (ans []int) {
n := len(security)
left := make([]int, n)
right := make([]int, n)
//求解left与right数组
for i:=1; i<n; i++ {
if security[i] >= security[i-1] {
left[i] = left[i-1] + 1
}
if security[n-i] >= security[n-i-1] {
right[n-i-1] = right[n-i] + 1
}
}
//求解适合打劫的日子
for i:=0; i<n; i++ {
if left[i]>=time && right[i]>=time {
ans = append(ans, i)
}
}
return
}
