AC
思路:
- 建立一个数组a[i]存输入数,再建立一个数组b[i]存a[i]的下一个最大值
- 维护一个单调递增栈
- 让a[i]倒序读入,此时栈空,第一个元素下一个最大值必为-1;若栈内元素比他大,说明为他的下一个最大值,存入b[i]中,然后a[i]压栈;
- 若栈内元素比他大或等于他,则栈内元素出栈,直至到达比他大的数,若没有,则为-1,若有,存入b[i]中,然后a[i]压栈;
问题:
1. 当数据量超过1e5的时候,不用cin cout,会超时!!!
2. 建立一个数组a[i]存输入数,再建立一个数组b[i]存a[i]的下一个最大值,不需要b[a[i]],会超时
#include <iostream>
#include <cstdio>
#include<stack>
using namespace std;
int a[100010];
int b[100010];
int main(){
int x,i;
scanf("%d",&x);
while(x--){
int n;
scanf("%d",&n);
stack<int> s;
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
}
for(i=n;i>=1;i--){
if(s.empty()) {
s.push(a[i]);
b[i]=-1;
}
else{
if(a[i]<s.top()){
b[i]=s.top();
s.push(a[i]);
}
else{
while(!s.empty()&&a[i]>=s.top())
s.pop();
if(s.empty()){
b[i]=-1;
s.push(a[i]);
}
else{
b[i]=s.top();
s.push(a[i]);
}
}
}
}
for(i=1;i<=n;i++){
printf("%d-->%d\n",a[i],b[i]);
}
if(x) printf("\n");
}
return 0;
}