数据结构实验之栈与队列六:下一较大值(二)
Description
对于包含n(1<=n<=100000)个整数的序列,对于序列中的每一元素,在序列中查找其位置之后第一个大于它的值,如果找到,输出所找到的值,否则,输出-1。
Input
输入有多组,第一行输入t(1<=t<=10),表示输入的组数;
以后是 t 组输入:每组先输入n,表示本组序列的元素个数,之后依次输入本组的n个元素。
Output
输出有多组,每组之间输出一个空行(最后一组之后没有);
每组输出按照本序列元素的顺序,依次逐行输出当前元素及其查找结果,两者之间以-->间隔。
Samples
Sample #1
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 <bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"
const int N = 2e5 + 10;
const int INF = 1e18;
const int mod = 1e9 + 7;
int n, k, ads, m, t, x, ans, a[N], h[N];
string s[3000];
void ClearFloat()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
}
signed main()
{
ClearFloat();
cin >> t;
while (t--)
{
stack<int>st;
cin >> n;
a[0] = -1;
for (int i = 1; i <= n; i++) cin >> a[i];
for (int i = n; i >= 1; i--)//倒着来,方便使单调栈为单调递增
{
while (st.size() && a[i] >= a[st.top()]) st.pop();//使得单调栈内元素位单调递增,这样队首即为下一个最大数
if (st.empty()) h[i] = 0;
else h[i] = st.top();
st.push(i);
}
for (int i = 1; i <= n; i++) cout << a[i] << "-->" << a[h[i]] << endl;
}
}