数据结构实验之栈与队列六：下一较大值（二）

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 <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
int main()
{
    int t;
    int n;
    int x;
    int a[100005];
    stack<int>st1;
    stack<int>st2;
    while(~scanf("%d",&t))
    {
        while(t--)
        {
        //清空栈。
            while(!st1.empty())
                st1.pop();
            while(!st2.empty())
                st2.pop();
            scanf("%d",&n);
            for(int i=0; i<n; i++)
                scanf("%d",&a[i]);
            for(int i=n-1; i>=0; i--)
            {
            //空栈，下一最大值一定是-1.
                if(st1.empty())
                {
                    st2.push(-1);
                    st1.push(a[i]);
                }
             //出栈，直到为空栈或者找到比自己大的数停止。
                else
                {
                    while(!st1.empty()&&a[i]>=st1.top())
                    {
                        st1.pop();
                    }
                    //没有找到那个它，就是-1了，入栈。
                    if(st1.empty())
                    {
                        st1.push(a[i]);
                        st2.push(-1);
                    }
                    else
                    {
                        x = st1.top();
                        st2.push(x);
                        st1.push(a[i]);
                    }
                }
            }
            //把第二个栈按照出栈的顺序输出就行了。
            for(int i=0; i<n; i++)
            {
                printf("%d-->%d\n",a[i],st2.top());
                st2.pop();
            }
            //最后注意一下，最后一组没有空行
            printf(t==0?"":"\n");
        }
    }
    return 0;
}

第二种方法：这种方法就是牺牲空间换时间。使用C写的。

#include <stdio.h>
#include <stdlib.h>
int a[100005];//输入的数据
int b[100005];//栈，存储数据
int c[100005];//栈，存储索引
int d[100005];//最终输出的数据
int main()
{
    int t,n,i,j;
    scanf("%d",&t);
    while(t--)
    {
        j = -1;
        memset(d,-1,sizeof(d));
        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            if(j==-1||b[j]>=a[i])
            {
                b[++j] = a[i];
                c[j] = i;
            }
            else if(j>-1&&b[j]<a[i])
            {
                while(j>-1&&b[j]<a[i])
                {
                    d[c[j--]] = a[i];
                }
                b[++j] = a[i];
                c[j] = i;
            }
        }
        for(i=0;i<n;i++)
            printf(i==n-1?"%d-->%d\n\n":"%d-->%d\n",a[i],d[i]);
    }
    return 0;
}

精简版：在第二种方式的基础上简化
#include <stdio.h>
#include <stdlib.h>
int a[100005],b[100005],c[100005],d[100005];
int main()
{
    int t,n,i,j;
    scanf("%d",&t);
    while(t--)
    {
        j = -1;
        memset(d,-1,sizeof(d));
        scanf("%d",&n);
        for(i=0; i<n; i++)
        {
            scanf("%d",&a[i]);
            while(j>-1&&b[j]<a[i])
                d[c[j--]] = a[i];
            b[++j] = a[i];
            c[j] = i;
        }
        for(i=0; i<n; i++)
            printf(i==n-1?"%d-->%d\n\n":"%d-->%d\n",a[i],d[i]);
    }
    return 0;
}