首先尝试的暴力解题,但是时间超限了
我们可以从前往后遍历,维护一个单调递减栈:对于每个点,如果它的值大于栈顶下标表示的数组值,就依次弹出栈顶元素、并且置这些栈顶下标对应的答案为当前点的下标。如果值小于栈顶下标对应的数组值,就压栈。最后输出答案数组即可。
#include<iostream>
#include<cstdio>
#include<iomanip>
#include<stack>
#include<cstdlib>
#include <algorithm>
#include<string.h>
#include<math.h>
#define llu unsigned long long
using namespace std;
int main()
{
int n,k=0,top=0;
cin >> n ;
long long a[n+10];
int ans[n+5]={
0},st[n+5];
for(int i=1;i<=n;i++)
cin >> a[i] ;
for(int i=1;i<=n;i++)
{
while (top && a[st[top]] < a[i]) ans[st[top--]] = i;
st[++top] = i;
}
for(int i=1;i<=n;i++)
cout << ans[i] << " ";
return 0;
}
或者从后往前遍历,维持一个单调递减栈:对于每个点 arr[i] ,弹出对应的数组值小于等于它的栈顶下标,直到栈顶下标对应的元素值 > arr[i] ,栈顶下标就是答案。然后将这个元素的下标压栈:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e6 + 10;
int arr[maxn], ans[maxn], st[maxn], top = -1;
int main() {
int n;
scanf("%d", &n);
for (int i = 1; i <= n; ++i) scanf("%d", &arr[i]);
//从栈底到栈顶的下标对应的数组值(优先级)依次递减,
//后进先出,对应->第i个元素之后第一个大于a_i的元素的下标
for (int i = n; i > 0; --i) {
//将先进栈(后面的)且不大于arr[i]的值的坐标清除掉,
//这一过程就是从i+1往后寻找第一个大于a_i的元素下标
while (top != -1 && arr[st[top]] <= arr[i]) --top;
ans[i] = (top == -1 ? 0 : st[top]); //top=-1表示之后没有大于arr[i]的值
st[++top] = i; //存储arr的数组下标
}
for (int i = 1; i <= n; ++i) {
if (i == 1) printf("%d", ans[i]);
else printf(" %d", ans[i]);
}
return 0;
}