CF-Round #622-div2-C题

CF-Round #622-div2-C题

C1 & C2. Skyscrapers（hard easy version)

传送门

简单版本数据量1e3
hard版本数据量5e5

这里直接给的是hard版本的代码

这道题贪心，用单调栈，前缀和，后缀和

题目大意：要建n座摩天大厦。其中要求建成的大楼的高度成单调型或者峰型，并且不超过所给的值。（峰型解释为从第一个数开始递增然后到达最大的数，再递减）。像骆驼的驼峰，要求你设计每座大厦的楼层。使得总的楼数最多

我们只需要借助数据结构单调栈可以解决hard版本

首先，什么是单调栈？
单调栈分为单调递增栈和单调递减栈。
单调递增栈是出栈的序列为单调递增的
比如序列：8 7 5 9 3 2 10
我们入栈8（现在栈中有8）
7 < 8（栈顶），入栈（现在栈中 7， 8）（7为栈顶）
5 < 7 (栈顶）， 入栈（现在栈中5， 7， 8）（5为栈顶）
9 > 5（栈顶），出栈5； 8 > 5(出栈8）；8 > 5(出栈8）；9入栈（现在栈中9）（9为栈顶）
3 < 9 (栈顶），入栈（现在栈中3， 9）（3为栈顶）
2 < 3(栈顶），入栈（现在栈中2， 3， 9）（2为栈顶）
10 > 2(栈顶），出栈2；10 > 3,出栈3；10 > 9，出栈9；10入栈
当前栈中只有10

单调递减栈就是反过来即可（出栈序列为单调递减的）

我们用l[]数组记录当前i作为峰值的左边的前缀和
用r[]数组记录当前i作为峰值的右边的后缀和
再跑一遍数组，标记峰值的地方。（记录最大值最大层数）然后更新楼数即可

前缀和和后缀和先开始这里我出了点问题emmm

比如前缀和这里：
栈存储的是下标

我先开始if条件那里没有乘i；感觉没必要。
后来发现。用意是：让i前面的所有大厦都建成a[i]的高度。
如果栈不为空，就把栈顶下标后面到当前处理的下标之间的高度都建成a[i]高度。
如果不乘的话。。那就错了emmm.
后缀和一样的道理
就不多说啦~
昨天和前天都在补最近网课落下的作业。zg的作业真的多。。emm原先上他的计算机图形学也没这么多作业啊。。。

代码部分：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 5e5 + 10;

ll a[N];
ll l[N];
ll r[N];
ll ans;
ll id;
int n;
stack<int> st;

int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	for (int i = 1; i <= n; i++)
	{
		while (!st.empty() && a[st.top()] >= a[i])
		{
			st.pop();
		}
		if (st.empty())
		{
			l[i] = a[i] * i;
		}
		else
		{
			l[i] = l[st.top()] + a[i] * (i - st.top());
		}
		st.push(i);
	}
	while (!st.empty())
	{
		st.pop();
	}
	for (int i = n; i >= 1; i--)
	{
		while (!st.empty() && a[st.top()] >= a[i])
		{
			st.pop();
		}
		if (st.empty())
		{
			r[i] = a[i] * (n - i + 1);
		}
		else
		{
			r[i] = r[st.top()] + a[i] * (st.top() - i);
		}
		st.push(i);
	}
	for (int i = 1; i <= n; i++)
	{
		ll tmp = l[i] + r[i] - a[i];
		if (tmp > ans)
		{
			id = i;
			ans = tmp;
		}
	}
	for (int i = id - 1; i >= 1; i--)
	{
		a[i] = min(a[i], a[i + 1]);
	}
	for (int i = id + 1; i <= n; i++)
	{
		a[i] = min(a[i], a[i - 1]);
	}
	for (int i = 1; i <= n; i++)
	{
		cout << a[i] << " ";
	}
	cout << endl;
	return 0;
}