題目描述
對於包含n(1<=n<=100000)個整數的序列,對於序列中的每一元素,在序列中查找其位置之後第一個大於它的值,如果找到,輸出所找到的值,否則,輸出-1。
輸入
以後是 t 組輸入:每組先輸入n,表示本組序列的元素個數,之後依次輸入本組的n個元素。
輸出
每組輸出按照本序列元素的順序,依次逐行輸出當前元素及其查找結果,兩者之間以-->間隔。
之所以寫這篇博客主要是想記錄一下手寫的棧。
示例輸入
2
4 12 20 15 18
5 20 15 25 30 6
示例輸出
12-->20
20-->-1
15-->18
18-->-1
20-->25
15-->25
25-->30
30-->-1
6-->-1
<span style="font-weight: normal;">#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#define stack_init_size 101000
using namespace std;
int n;
typedef struct node //定義結構體
{
int num,id,next;
}Node;
struct Stack //封裝的棧
{
Node *base,*top; //棧底指針和棧頂指針
void clear() //棧的清空,使棧頂指針重新回到棧底
{
top = base;
}
void InitStack() //棧的初始化
{
base=(Node *)malloc(stack_init_size*sizeof(Node));
top=base;
}
Node GetTop() //得到棧頂元素
{
Node e;
e=*(top-1); //棧頂指針指的是棧尾元素的下一位
return e;
}
bool isempty() //判斷棧是否爲空
{
return top == base;
}
void Push(Node e) //插入棧頂元素
{
*top++=e;
}
void Pop() //刪除棧頂元素
{
top--;
}
}ss;
Node a[100010];
int main()
{
int t;
scanf("%d",&t);
ss.InitStack();
for(int k=1;k<=t;k++)
{
ss.clear(); //對於每組數據記得要清空
if(k>1)
printf("\n");
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].num);
a[i].id=i;
a[i].next=-1;
if(ss.isempty())
{
ss.Push(a[i]);
}
else
{
while(!ss.isempty())
{
Node b;
b=ss.GetTop();
if(b.num<a[i].num)
{
a[b.id].next=a[i].num;
ss.Pop();
}
else
break;
}
ss.Push(a[i]);
}
}
for(int i=1;i<=n;i++)
printf("%d-->%d\n",a[i].num,a[i].next);
}
return 0;
}</span>