數據結構:下一較大值(棧的思想)

例題：

Problem Description

對於包含n（1<=n<=100000）個整數的序列，對於序列中的每一元素，在序列中查找其位置之後第一個大於它的值，如果找到，輸出所找到的值，否則，輸出-1。

Input

輸入有多組，第一行輸入t（1<=t<=10），表示輸入的組數；
以後是 t 組輸入：每組先輸入n，表示本組序列的元素個數，之後依次輸入本組的n個元素。

Output

輸出有多組，每組之間輸出一個空行（最後一組之後沒有）；
每組輸出按照本序列元素的順序，依次逐行輸出當前元素及其查找結果，兩者之間以–>間隔。

Sample Input

2
4 12 20 15 18
5 20 15 25 30 6

Sample Output

12–>20
20–>-1
15–>18
18–>-1

20–>25
15–>25
25–>30
30–>-1
6–>-1

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int a[100001];
    int b[100001];
    stack <int> q;
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int len;
        scanf("%d",&len);
        for(int i=0;i<len;i++)
        {
            scanf("%d",&a[i]);
        }
        b[len-1]=-1;
        for(int i=len-2;i>=0;i--)
        {
            if(a[i+1]>a[i])
            {
                q.push(a[i+1]);
                b[i]=a[i+1];
            }
            else
            {
                while(!q.empty())
                {
                    if(q.top()>a[i])
                    {
                        b[i]=q.top();
                        break;
                    }
                    else
                    {
                        q.pop();
                    }
                }
                if(q.empty())
                {
                    b[i]=-1;
                }
            }
        }
        for(int i=0;i<len;i++)
        {
            printf("%d-->%d\n",a[i],b[i]);
        }
        if(n!=0)
        {
            printf("\n");
        }
    }
    return 0;
}