数据结构实验之栈六:下一较大值(二)

数据结构实验之栈六:下一较大值(二)

Time Limit: 150MS Memory Limit: 8000KB

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 <bits/stdc++.h>
using namespace std;


struct node
{
    int data;//当前数值
    int Id;//标记当前数值Id所在,若找到其后的较大值,更新其所在Id数值
    int nextdata;//下一较大值
};
node array[100010];


int main()
{
    int n, k;
    stack<node>S;
    while(scanf("%d",&k)!=EOF)
    {
        for(int t = 1; t <= k; t++)
        {
            while(!S.empty())
            {
                S.pop();
            }//栈清空
            scanf("%d",&n);
            for(int j = 1; j <= n; j++)
            {
                scanf("%d",&array[j].data);
                array[j].Id = j;  //初始化
                array[j].nextdata = -1;
                if(S.empty()) //栈内无元素,第一个进栈
                {
                    S.push(array[j]);
                }
                else //栈不为空
                {
                    while(!S.empty())//不断取其栈顶元素,更新其栈顶元素下一较大值
                    {
                        node tmp = S.top();//取栈顶元素 ,从第一个元素开始
                        int k = tmp.Id;//获取栈顶元素Id
                        if(tmp.data < array[j].data)//将当前元素与其上一个元素数据比较大小
                        {
                            array[k].nextdata = array[j].data;//更新栈顶元素Id所在的较大值
                            S.pop();//将较大值出栈
                        }
                        else break;
                    }
                    S.push(array[j]);//将当前数值入栈,便于查找当前值下一较大值
                }
            }
            for(int i = 1; i <= n; i++)
            {
                printf("%d-->%d\n",array[i].data,array[i].nextdata);
            }
            if(t != k)
            {
                printf("\n");
            }
        }
    }
    return 0;
}

原文地址:https://www.cnblogs.com/CCCrunner/p/11782143.html