[Leetcode456]132模式 - 单调栈
给你一个整数数组 nums ,数组中共有 n 个整数。132 模式的子序列 由三个整数 nums[i]、nums[j] 和 nums[k] 组成,并同时满足:i < j < k 和 nums[i] < nums[k] < nums[j] 。
-
暴力遍历,O3的复杂度,必超时。
-
为什么用单调栈: 1 - 3 - 2 模式,在数组中,有局部单调递增递减趋势,都可以使用单调栈。
-
从后往前遍历,为什从后往前遍历,k值是中间量?
思路:维护j栈,因为 j 比 k大,因此还是维护一个递减栈,
以 3 2 4 5 6 4 7(3 2 5 6 4 5)为例子, stack_k 7->4 遇到了6,发现6比4 大,则4 出栈作为k,6填入作为j,
继续维护j栈, 6 -> 5 -> 4 -> 2,走到这里可以发现,已经满足条件,完成遍历。 -
从解题思路来说,对于局部递增递减序列来说,算法最好的切入点在于中间量,可以减少大量判断语句。
bool find132pattern(int* nums, int numsSize)
{
int i, j, k;
int *stack = (int *)calloc(numsSize, sizeof(int));
int stack_top = 0;
if (numsSize < 3) {
return false;
}
i = nums[numsSize - 3];
j = nums[numsSize - 2];
k = nums[numsSize - 1];
if (j > i && j > k && k > i) {
return true;
}
stack[0] = nums[numsSize - 1];
k = stack[0];
for (int index = numsSize - 2; index >= 0; index--) {
if (nums[index] < stack[0] && nums[index] < k && stack[0] > k) {
return true;
}
while (nums[index] > stack[stack_top]) {
k = stack[stack_top];
stack_top--;
if (stack_top == -1) {
break;
}
}
stack_top++;
stack[stack_top] = nums[index];
}
return false;
}