Space is cheap. Time, isn't - Part 2
Another problem where we trade space for time. The idea is to calculate the max decreasing sequence per element from the right and from the left, for a total of O(2n) space, in linear time. During the second loop, we also check if the element meets the "time" criteria. Space is cheap, time, isn't. Code is down below. To honor the 80th anniversary of the Pearl Harbor attack, here is a picture that I took on the Battleship Missouri, in Pearl Harbor, this year (2021). Cheers, ACC.
Find Good Days to Rob the Bank - LeetCode
You and a gang of thieves are planning on robbing a bank. You are given a 0-indexed integer array security
, where security[i]
is the number of guards on duty on the ith
day. The days are numbered starting from 0
. You are also given an integer time
.
The ith
day is a good day to rob the bank if:
- There are at least
time
days before and after theith
day, - The number of guards at the bank for the
time
days beforei
are non-increasing, and - The number of guards at the bank for the
time
days afteri
are non-decreasing.
More formally, this means day i
is a good day to rob the bank if and only if security[i - time] >= security[i - time + 1] >= ... >= security[i] <= ... <= security[i + time - 1] <= security[i + time]
.
Return a list of all days (0-indexed) that are good days to rob the bank. The order that the days are returned in does not matter.
Example 1:
Input: security = [5,3,3,3,5,6,2], time = 2 Output: [2,3] Explanation: On day 2, we have security[0] >= security[1] >= security[2] <= security[3] <= security[4]. On day 3, we have security[1] >= security[2] >= security[3] <= security[4] <= security[5]. No other days satisfy this condition, so days 2 and 3 are the only good days to rob the bank.
Example 2:
Input: security = [1,1,1,1,1], time = 0 Output: [0,1,2,3,4] Explanation: Since time equals 0, every day is a good day to rob the bank, so return every day.
Example 3:
Input: security = [1,2,3,4,5,6], time = 2 Output: [] Explanation: No day has 2 days before it that have a non-increasing number of guards. Thus, no day is a good day to rob the bank, so return an empty list.
Example 4:
Input: security = [1], time = 5 Output: [] Explanation: No day has 5 days before and after it. Thus, no day is a good day to rob the bank, so return an empty list.
Constraints:
1 <= security.length <= 105
0 <= security[i], time <= 105
public IListGoodDaysToRobBank(int[] security, int time) { int[] maxDecreasingSequenceRight = new int[security.Length]; int[] maxDecreasingSequenceLeft = new int[security.Length]; for (int i = security.Length - 2; i >= 0; i--) maxDecreasingSequenceRight[i] = (security[i] <= security[i + 1]) ? (maxDecreasingSequenceRight[i + 1] + 1) : 0; List retVal = new List (); for (int i = 0; i < security.Length; i++) { maxDecreasingSequenceLeft[i] = (i > 0 && security[i] <= security[i - 1]) ? (maxDecreasingSequenceLeft[i - 1] + 1) : 0; if (maxDecreasingSequenceLeft[i] >= time && maxDecreasingSequenceRight[i] >= time) { retVal.Add(i); } } return retVal; }
Comments
Post a Comment