算法培训 Day 1-2 单调栈

算法培训 Day 1-2 单调栈

给出一些数 定义函数 f(i) 代表数列中第 i个元素之后第一个大于a[i]的元素的下标

输出每个数的 f(i)

解答

维护一个单调栈 栈内的元素直接为 数组数据的下标 i 然后每到下一个数 就和当前栈顶的指向的元素进行比较 如果小于就放入，如果大于则一直出栈 直到栈顶大于当前元素

因此可以栈栈内先存入一个无穷大的数 的下标，设a[0]=inf 然后把0 存入栈内

	a[0]=inf;
	//a[n+1]=inf;
	s.push(0);
	i=1;
	while(i<=n)
	{
		while(a[s.top()]<a[i])
		{
			b[s.top()] = i;
            //b 数组存放第一个大于 a[s.top()]元素的下标  即 f(i)
			s.pop();
			if(s.empty())
				break;
		}
		s.push(i);
		i++;
	}

上面即是核心思想，然后下面是题目源码

#include<bits/stdc++.h>
using namespace std;
const int Maxn = 3000000+300;
long long a[Maxn];
long long b[Maxn];
const int inf = 1e9+100;
int main()
{
	int n;
	int i;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%lld",&a[i]);
		
	stack<int> s;
	a[0]=inf;
	//a[n+1]=inf;
	s.push(0);
	i=1;
	while(i<=n)
	{
		while(a[s.top()]<a[i])
		{
			b[s.top()] = i;
			s.pop();
			if(s.empty())
				break;
		}
		s.push(i);
		i++;
	}
	for(i=1;i<n;i++)
		printf("%d ",b[i]);
	printf("0");
		
	return 0;
}