数据结构实验之栈与队列六:下一较大值(二)
数据结构实验之栈与队列六:下一较大值(二)
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 Output12–>20 20–>-1 15–>18 18–>-1
20–>25 15–>25 25–>30 30–>-1 6–>-1 Hint 本题数据量大、限时要求高,须借助栈来完成。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#define INF 0x3f3f3f3f
#define maxsize 110000
using namespace std;
int n;
int a[110000],b[110000];
typedef struct
{
int *elem;
int top,len;
} stack;
void creat(stack *s)
{
s->elem=(int *)malloc(maxsize*sizeof(int));
s->top=-1;
s->len=n;
}
void push(stack *s,int x)
{
s->elem[++s->top]=x;
}
void pop(stack *s)
{
s->top--;
}
int front(stack *s)
{
return s->elem[s->top];
}
int empty(stack *s)
{
if(s->top==-1) return 1;
return 0;
}
int main()
{
int t;
stack s;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
creat(&s);
for(int i=0; i<n; i++)
scanf("%d",&a[i]);
b[n-1]=-1;
push(&s,a[n-1]);
for(int i=n-2; i>=0; i--)
{
if(a[i]>front(&s))
{
while(!empty(&s)&&a[i]>=front(&s))
pop(&s);
if(empty(&s))
b[i]=-1;
else if(a[i]<front(&s))
b[i]=front(&s);
push(&s,a[i]);
}
else
{
b[i]=front(&s);
push(&s,a[i]);
}
}
for(int i=0; i<n; i++)
printf("%d-->%d\n",a[i],b[i]);
if(t) printf("\n");
}
return 0;
}