数据结构实验之栈与队列六:下一较大值(二)
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
Output
12–>20
20–>-1
15–>18
18–>-1
20–>25
15–>25
25–>30
30–>-1
6–>-1
Hint
本题数据量大、限时要求高,须借助栈来完成。
#include <stdio.h>
#include <stdlib.h>
int main()
{
int i, t, n, a[100010], b[100010], s[100010],c;//数组a用来存储input数据,b用来存储较大值或-1,s为栈
scanf("%d", &t);
while (t--)
{
scanf("%d", &n);
for (i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
}
c = 0;
b[n] = -1;//最后一个元素后面没有数据,所以输出-1即可,把-1先存入数组b的最后一个值
s[++c] = a[n];//因为a[n]不会和它后面的数比较,所以先将它存入栈底与它前面的一个数进行比较
for (i = n - 1; i >= 1; i--)//由于栈思想,所以数据是由后往前
{
s[++c] = a[i];//新数先存入栈,作为栈顶,它下面的数都是它后面的数
while (a[i] >= s[c] && c > 0)//删除想找的数值的后面比它小的数
{
c--;
}
if (c == 0)//栈空即为c=0
{
b[i] = -1;
}
else//c!=0,也就是后面还有比想找的数大的数
{
b[i] = s[c];
}
s[++c] = a[i];//进行找数操作后,将当前想找的数再存入栈中,保证下一个数的后面的数据完整性
}
for (i = 1; i <= n; i++)
{
printf("%d-->%d\n", a[i], b[i]);
}
if (t > 0)
{
printf("\n");
}
}
}
在寻找20后面第一个大于20的数时,没有找到所以栈被清空,所以20比其后面的数都大,在找12后面比12大的数时只剩下20这一个数,比12大,20就可,有了20这种数,即可清空栈,删除对于前面的数没有用的的数,节省了时间