栈具有先进后出的性质,单调栈就是利用一些巧妙的逻辑,使得栈内的元素都保持有序(单调递增或单调递减)
根据每日气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。
如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
维护递减栈,后入栈的元素总比栈顶元素小。比对当前元素与栈顶元素的大小
class Solution:
def dailyTemperatures(self, T: List[int]) -> List[int]:
res = [0] * len(T)
min_stack = [] # 最小栈,维持一个递减的栈 ,栈记录的是天数
for i in range(len(T)):
# 如果栈不为空,且当前元素大于栈顶元素
while min_stack!=[] and T[i]>T[min_stack[-1]]:
idx = min_stack.pop()
res[idx] = i - idx
min_stack.append(i)
return res
用栈记录天数和当天的气温
class Solution(object):
def dailyTemperatures(self, T):
"""
:type T: List[int]
:rtype: List[int]
"""
if len(T)==1:
return [0]
res = [0] * len(T)
min_stack = [] # 最小栈,维持一个递减的栈
for i in range(len(T)):
if min_stack==[] or T[i]<=min_stack[-1][1]:
min_stack.append((i, T[i]))
else:
while min_stack!=[] and T[i]>min_stack[-1][1]:
temp = min_stack.pop()
res[temp[0]] = i - temp[0]
min_stack.append((i, T[i]))
return res
给定两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
示例 1:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。
对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。
对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。
示例 2:
输入: nums1 = [2,4], nums2 = [1,2,3,4].
输出: [3,-1]
解释:
对于 num1 中的数字 2 ,第二个数组中的下一个较大数字是 3 。
对于 num1 中的数字 4 ,第二个数组中没有下一个更大的数字,因此输出 -1 。
提示:
nums1和nums2中所有元素是唯一的。
nums1和nums2 的数组大小都不超过1000。
先单独处理nums2,将nums2中每一个数的下一个最大元素都找出来,用哈希表保存;
然后对于nums1中的每一个元素再在哈希表中找对应的,不存在的就为-1
# 复杂版
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
res = [-1] * len(nums1)
stack_map = {}
stack = []
for i in range(len(nums2)):
if stack==[] or nums2[i]<=stack[-1]:
stack.append(nums2[i])
else:
while stack!=[] and nums2[i]>stack[-1]:
temp = stack.pop()
stack_map[temp] = nums2[i]
stack.append(nums2[i])
for i in range(len(nums1)):
if nums1[i] in stack_map:
res[i] = stack_map[nums1[i]]
return res
# 精简版
class Solution(object):
def nextGreaterElement(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: List[int]
"""
stack, hash_map = [], {}
for n in nums2:
while stack!=[] and stack[-1]<n:
hash_map[stack.pop()] = n
stack.append(n)
return [hash_map.get(x, -1) for x in nums1]
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。
数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,
这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
将两个数组拼接起来,其余地和前面一样
class Solution(object):
def nextGreaterElements(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
单调栈
"""
res = [-1] * len(nums)
n = len(nums)
nums += nums
min_stack = []
for i in range(len(nums)):
if min_stack==[] and i>=n:
break
while min_stack!=[] and nums[min_stack[-1]]<nums[i]:
res[min_stack.pop()] = nums[i]
# 如果这个还是第一个数组中的元素就可以入栈
if i<n:
min_stack.append(i)
return res
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
int len_num = nums.size();
vector<int> res(len_num, -1);
// nums.insert(nums.end(), nums.begin(), nums.end());
vector<int> min_stack;
for (int i=0; i<len_num*2; ++i) {
// 不用拼接数组,直接取余也可以
while (!min_stack.empty() && nums[min_stack.back()]<nums[i%len_num]) {
res[min_stack.back()] = nums[i%len_num];
min_stack.pop_back();
}
if (i<len_num) min_stack.push_back(i);
}
return res;
}
};
给定一个32位正整数 n,你需要找到最小的32位整数,其与 n 中存在的位数完全相同,并且其值大于n。
如果不存在这样的32位整数,则返回-1。
示例 1:
输入: 12
输出: 21
示例 2:
输入: 21
输出: -1
class Solution(object):
def nextGreaterElement(self, n):
"""
:type n: int
:rtype: int
"""
nums = list(map(int, str(n)))
stack = [nums.pop()]
# 从右往左找升序,
while nums and stack[-1] <= nums[-1]:
stack.append(nums.pop())
if not nums: return -1
i = len(nums)-1
# 直到找到 第一个大于i的元素
# print(stack)
for j in range(len(stack)):
if stack[j] > nums[-1]:
stack[j], nums[-1] = nums[-1], stack[j]
break
nums += stack
res = "".join(map(str,nums))
return -1 if int(res)>(1<<31)-1 else res