數據結構實驗之棧六：下一較大值（二）

數據結構實驗之棧六：下一較大值（二）

Time Limit: 150MS Memory Limit: 8000KB

Submit Statistic

Problem Description

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

Input

 輸入有多組，第一行輸入t（1<=t<=10），表示輸入的組數；

以後是 t 組輸入：每組先輸入n，表示本組序列的元素個數，之後依次輸入本組的n個元素。

Output

 輸出有多組，每組之間輸出一個空行（最後一組之後沒有）；

每組輸出按照本序列元素的順序，依次逐行輸出當前元素及其查找結果，兩者之間以-->間隔。

Example Input

24 12 20 15 185 20 15 25 30 6 

Example Output

12-->2020-->-115-->1818-->-120-->2515-->2525-->3030-->-16-->-1

Hint

 本題數據量大、限時要求高，須藉助棧來完成。

#include<stdio.h>
#include<malloc.h>
#include<stdlib.h>

#define maxsize 100001
typedef struct
{
    int *top;//用來存放元素的值
    int *id;//存放數據的座標
    int *base;
}qstack;

struct node//定義結構體存放數據
{
    int data;
    int next;
}boy[100001];

//初始化棧
void Initstack(qstack &s)
{
    s.base=(int *)malloc(maxsize*sizeof(int));//定義的棧中有三個變量top,id,base;不能只開一次內存
    s.top=s.base;
    s.id=(int *)malloc(maxsize*sizeof(int));
}

//進棧並判斷
void Comstack(qstack &s,struct node[],int n)
{
    for(int i=0;i<n;i++)
    {
        scanf("%d",&boy[i].data);
        boy[i].next=-1;//將元素的下一較大值初始化爲-1
        if(s.top==s.base)//棧頂等於棧底的時候直接將元素和元素的位置座標入棧
        {
            *(++s.top)=boy[i].data;
            *(++s.id)=i;
        }
        else
        {
            if(*(s.top)>=boy[i].data)//棧頂大於當前元素的時候直接將元素和元素的位置座標入棧
            {
                *(++s.top)=boy[i].data;
                *(++s.id)=i;
            }
            else if(*(s.top)<boy[i].data)//棧頂元素小於等於當前元素的時候
            {
                while((*(s.top)<boy[i].data)&&(s.top!=s.base))//直到找到不再小於當前元素或棧頂等於棧底的時候推出
                {
                    boy[*s.id].next=boy[i].data;//當前元素爲站定元素的下一較大值
                    s.top--;//退棧
                    s.id--;
                }
                *(++s.top)=boy[i].data;//將新的元素的值和位置座標入棧
                *(++s.id)=i;
            }
        }
    }
}

//輸出結果
void Putstack(struct node[],int n)
{
    for(int i=0;i<n;i++)
    {
        printf("%d-->%d\n",boy[i].data,boy[i].next);
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    int n;
    while(t--)
    {
        scanf("%d",&n);
        qstack s;
        Initstack(s);
        Comstack(s,boy,n);
        Putstack(boy,n);
        printf("\n");
    }
    return 0;
}