数据结构实验之栈六:下一较大值(二)
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 num,next,id;
};
void nextmax(int n)
{
stack<struct node>s;
struct node a[100050],t;
for(int i=1; i<=n; i++)
{
scanf("%d",&a[i].num); //输入数据
a[i].next=-1; //初始化置为-1
a[i].id=i; //下标
if(s.empty()) //栈为空
s.push(a[i]); //压入栈
else //栈不为空
{
while(!s.empty())
{
t=s.top(); //栈顶
if(a[i].num>t.num) //大于栈顶
{
a[t.id].next=a[i].num;
s.pop(); //出栈
}
else
break;
}
s.push(a[i]); //压入栈
}
}
for(int i=1; i<=n; i++)
printf("%d-->%d\n",a[i].num,a[i].next);
}
int main()
{
int n,T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
nextmax(n);
if(T!=0)
printf("\n");
}
return 0;
}