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

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
Hint
本题数据量大、限时要求高，须借助栈来完成。

题解：
本题借助栈来完成，由于栈有后进先出的特点因此我在做题的时候用了5个栈（可能有点多），其中s栈是用来操作的，p栈是为了存储数据，由于刚存入的数据是逆序的因此需要q栈来将数据变成顺序用来输出。x栈是用来存储找到的第一个比这个元素大的元素，y栈是用来将比较大的元素存储起来。题目要求对于序列中的每一元素，在序列中查找其位置之后第一个大于它的值，我们可以从后面开始，最后一个元素因为后面没有元素了，所以对应的大于它的没有，因此是-1，从最后一个元素开始往前走，如果碰到这个元素大于它前面的元素，那么后面的元素就是前面的元素的第一个大于它的值，将这个大的元素存入x栈，同时存入y栈，再往前走，如果这个元素不大于它前一个元素，则说明这个元素不是它前面的那个元素的第一个大于它的值，那么它前面的那个元素的第一个大于它的值可能在后面，也可能不存在，刚才我们用y栈存下了大的元素，因此只要在y栈中寻找即可，若找到了就停止，没找到就让y中的元素出栈，直到y中没有元素，那么就说明序列中没有比这个元素大的，所以大于它的值为-1。就这样依次寻找，直到重复到第一个元素，就完成了。

#include <stdio.h>
#include <stdlib.h>
#define stackinitsize 110000
#define stackcreat 10
#define ok 1
#define error -1
#define overflow -1
typedef int elemtype;
typedef struct
{
    elemtype *base;
    elemtype *top;
    int stacksize;

} sqstack;
int initstack(sqstack *s)
{
    s->base=(elemtype*)malloc(stackinitsize*sizeof(elemtype));
    if(!s->base)exit(overflow);
    s->top=s->base;
    s->stacksize=stackinitsize;
    return ok;

}
int push(sqstack *s,elemtype e)
{
    if(s->top-s->base>=s->stacksize)
    {
        s->top=(elemtype*)realloc(s->base,(s->stacksize+stackcreat)*sizeof(elemtype));
        if(!s->base)exit(overflow);
        s->top=s->base+s->stacksize;


    }*(s->top++)=e;
    return ok;

}
int pop(sqstack *s)
{
    if(s->top==s->base)return error;
    s->top--;
    return ok;

}
int gettop(sqstack *s)
{
    int e;
    if(s->top==s->base)return error;
    e=*(s->top-1);
    return e;

}
int stackempty(sqstack *s)
{
    if(s->top==s->base)return 1;
    else return 0;

}

int main()
{
    int t,k,n,e1,e2,e,i,a;
    sqstack s,p,q,x,y;
    initstack(&s);
    initstack(&p);
    initstack(&q);
    initstack(&x);
    initstack(&y);
    scanf("%d",&t);
    for(k=1; k<=t; k++)
    {
        scanf("%d",&n);
        for(i=1; i<=n; i++)
        {
            scanf("%d",&a);
            push(&s,a);//把元素存到s栈
            push(&p,a);//把元素存到p栈

        }
        while(!stackempty(&p))//把p栈中的元素转到q栈，让元素输出的时候是顺序的
        {
            e=gettop(&p);
            push(&q,e);
            pop(&p);

        }
        push(&x,-1);//最后一个元素比它大的数是-1
        while(!stackempty(&s))
        {
            int f=0;
            e1=gettop(&s);
            pop(&s);
            if(stackempty(&s))break;
            else e2=gettop(&s);
            if(e1>e2)
            {
                push(&x,e1);
                push(&y,e1);

            }
            else
            {
                while(!stackempty(&y))//从y栈中找
                {
                    e=gettop(&y);
                    if(e>e2)
                    {
                        push(&x,e);
                        f=1;
                        break;

                    }
                   else  pop(&y);

                }
                if(f==0)push(&x,-1);//没找到，则大的是-1

            }

        }
        while(!stackempty(&x)&&!stackempty(&q))
        {
            e1=gettop(&q);
            e2=gettop(&x);
            printf("%d-->%d\n",e1,e2);
            pop(&x);
            pop(&q);

        }
        if(k!=t)printf("\n");

    }

    return 0;
}

参考代码二：（用数组做的）

#include <stdio.h>
#include <stdlib.h>
int a[110000],b[110000],c[110000];
int main()
{

    int n,i,j,t,k,x,y;
    scanf("%d",&t);
    for(k=1; k<=t; k++)
    {
        scanf("%d",&n);
        for(i=1; i<=n; i++)scanf("%d",&a[i]);
        b[n]=-1;
        x=n-1;
        y=0;
        for(j=n; j>=1; j--)
        {
            int f=0;
            if(a[j]>a[j-1])
            {
                b[x--]=a[j];
                c[++y]=a[j];

            }
            else
            {
                while(y>=1)
                {
                    if(c[y]>a[j-1])
                    {
                        b[x--]=c[y];
                        f=1;
                        break;
                    }
                    else  y--;

                }
                if(f==0)b[x--]=-1;

            }

        }
        for(i=1; i<=n; i++)
            printf("%d-->%d\n",a[i],b[i]);
        if(k!=t)printf("\n");
    }
    return 0;
}