今天继续总结两道有关栈的题目
第一题:每日温度
这道题目的话,采用暴力遍历是会超时的,所以我们采用栈来解决题目。
for 循环要从后往前扫描元素,倒着入栈,正着出栈。
while 循环是把两个较大元素之间的元素排除
部分代码:
int* dailyTemperatures(int* T, int TSize, int* returnSize){
*returnSize = TSize;
int *ans = malloc(sizeof(int)*TSize);
int stack[TSize+1]; //栈底不放元素,因此要多一个空间
int top = 0;
for(int i = TSize-1; i >= 0; i--) //for 循环要从后往前扫描元素{
while(top != 0 && T[stack[top]] <= T[i])//如果在栈中找不到比T[i]大的元素,则栈减,直至找到比T[i]大的元素或者top为0
top--;
ans[i] = top==0 ? 0 : (stack[top] - i);//利用下标得到索引间距,若top=0,说明栈中(即该元素后)没有比当前元素大的。
stack[++top] = i; //将下标放入栈
}
return ans;
}
第二题:下一个更大元素 II
借助官方题解,我们可以使用单调栈来解决这个问题。
我们首先把第一个元素 A[1] 放入栈,随后对于第二个元素 A[2],如果 A[2] > A[1],那么我们就找到了 A[1] 的下一个更大元素 A[2],此时就可以把 A[1] 出栈并把 A[2] 入栈;如果 A[2] <= A[1],我们就仅把 A[2] 入栈。对于第三个元素 A[3],此时栈中有若干个元素,那么所有比 A[3] 小的元素都找到了下一个更大元素(即 A[3]),因此可以出栈,在这之后,我们将 A[3] 入栈,以此类推。
可以发现,我们维护了一个单调栈,栈中的元素从栈顶到栈底是单调不降的。当我们遇到一个新的元素 A[i] 时,我们判断栈顶元素是否小于 A[i],如果是,那么栈顶元素的下一个更大元素即为 A[i],我们将栈顶元素出栈。重复这一操作,直到栈为空或者栈顶元素大于 A[i]。此时我们将 A[i] 入栈,保持栈的单调性,并对接下来的 A[i + 1], A[i + 2] … 执行同样的操作。
由于这道题的数组是循环数组,因此我们需要将每个元素都入栈两次。这样可能会有元素出栈找过一次,即得到了超过一个“下一个更大元素”,我们只需要保留第一个出栈的结果即可。
int* nextGreaterElements(int* nums, int numsSize, int* returnSize){
if(numsSize==1){nums[0]=-1;*returnSize=1;return nums;}//个数为1的特殊情况
if(numsSize==0){*returnSize=0;return nums;}//个数为0的特殊情况
int*stack=(int*)malloc(sizeof(int)*(numsSize*2+1));//我们要保证每个元素入栈2次
int*ans=(int*)malloc(sizeof(int)*numsSize);
memset(ans,-1,sizeof(ans));
memset(stack,-1,sizeof(stack));
int top=-1,x=numsSize-1;
for(int i=numsSize-1;i>=0;i--){//第一次从后向前遍历
while(top!=-1&&stack[top]<nums[i])top--;
if(top==-1){
stack[++top]=nums[i];
ans[i]=-1;
}
else{
ans[i]=stack[top];
stack[++top]=nums[i];
}
}
for(int i=numsSize-1;i>=0;i--){//第二次从后向前遍历
while(top!=-1&&stack[top]<=nums[i])top--;
if(top==-1){
stack[++top]=nums[i];
ans[i]=-1;
}
else{
ans[i]=stack[top];
stack[++top]=nums[i];
}
}
*returnSize=numsSize;
free(stack);
return ans;
}