数据结构实验之栈与队列六:下一较大值(二)
Time Limit: 150MS
Memory Limit: 8000KB
Problem Description
对于包含n(1<=n<=100000)个整数的序列,对于序列中的每一元素,在序列中查找其位置之后第一个大于它的值,如果找到,输出所找到的值,否则,输出-1。
Input
输入有多组,第一行输入t(1<=t<=10),表示输入的组数;
以后是 t 组输入:每组先输入n,表示本组序列的元素个数,之后依次输入本组的n个元素。
Output
输出有多组,每组之间输出一个空行(最后一组之后没有);
每组输出按照本序列元素的顺序,依次逐行输出当前元素及其查找结果,两者之间以-->间隔。
Example Input
2 4 12 20 15 18 5 20 15 25 30 6
Example Output
12-->20 20-->-1 15-->18 18-->-1 20-->25 15-->25 25-->30 30-->-1 6-->-1
Hint
本题数据量大、限时要求高,须借助栈来完成。
Author
#include <stdio.h> #include <stdlib.h> #include <string.h> int s[400001],a[400000],b[400000],c; int main() { int i,j,top,n,head,t,k,len; scanf("%d",&t); for(k=1; k<=t; k++) {memset(b,0,sizeof(b)); memset(s,0,sizeof(s)); memset(a,0,sizeof(a)); scanf("%d",&n); for(i=1; i<=n; i++) scanf("%d",&a[i]); top=0; b[n]=-1; s[++top]=a[n]; for(i=n-1; i>=1; i--) { if(a[i]<s[top]) { b[i]=s[top]; s[++top]=a[i]; } else { while(a[i]>=s[top]&&top!=0) top--; if(top==0) { b[i]=-1;s[++top]=a[i]; } else {b[i]=s[top]; s[++top]=a[i];} } } for(i=1; i<=n; i++) printf("%d-->%d\n",a[i],b[i]); if(k!=t) printf("\n"); } return 0; }
从后往前,最后一个为栈顶,如果a【i】<栈顶则a【i】进栈,成为栈顶,否则循环找到在栈里比a【i】大的数,一次次删掉栈顶,如果找不到则a【i】成为栈中的唯一值栈顶,找到则a【i】也成为栈顶,与栈中数形成递增数列;
要设一个数组一直保存a【i】的对应值;