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
本题数据量大、限时要求高,须借助栈来完成。
Source
#include <stdio.h>
#include <stdlib.h>
#define MAXSIZE 100001
#define ERROR -9999999
int main()
{
int n;
int a[MAXSIZE]; // 原数据,顺序储存
int b[MAXSIZE]; // 较大数集合
int s[MAXSIZE]; // 栈,用于存储递增序列
int top = -1;
int i;
int m;
while(~scanf("%d", &m))
{
while(m--)
{
scanf("%d", &n);
for (i = 0; i < n; i++)
{
scanf("%d", &a[i]);
}
i = n-1;
b[i] = -1; // 最后一个数的较大数一定为 -1
for (i = n-2; i >= 0; i--) // 倒序遍历原数据
{
int flag = 0;
if (a[i+1] > a[i]) // 若a[i+1] > a[i], 则a[i+1]入栈, 且当前数的较大数为a[i+1]
{
top++;
s[top] = a[i+1];
b[i] = a[i+1];
}
else // 若a[i+1] <= a[i], 则需要在栈中寻找较大数
{
while(top != -1)
{
if (a[i] < s[top]) // 若在栈中找到较大数,则当前数的较大数为栈顶元素
{
b[i] = s[top];
flag = 1;
break;
}
top--; // 将比此元素小的所有元素出栈
}
if (flag == 0) // 若未找到较大数,则当前数的较大数为-1
{
b[i] = -1;
}
}
}
for (i = 0; i < n; i++)
{
printf("%d-->%d
", a[i], b[i]);
}
if (m != 0)
{
printf("
");
}
}
}
return 0;
}