1.13学习博客

今天继续总结两道有关栈的题目
第一题：每日温度

这道题目的话，采用暴力遍历是会超时的，所以我们采用栈来解决题目。

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;
}