2021牛客暑期多校训练营2 K stack（思维 + 构造）

最新推荐文章于 2022-08-03 21:24:09 发布

原创最新推荐文章于 2022-08-03 21:24:09 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

思维同时被 2 个专栏收录

16 篇文章

订阅专栏

牛客多校

7 篇文章

订阅专栏

本文介绍如何利用给定的栈大小数组bbb和入栈元素数组aaa，通过构建合法栈并遵循特定规则来模拟栈操作。核心思路是构造bbb数组并检查其合法性，然后使用单调栈实现元素的入栈和出栈。

K stack

题目大意：
给定数组 $b$ 中的 $k$ 个元素，其记录了栈的大小，而数组 $a$ 存储入栈的元素，当栈顶元素大于入栈元素时则弹出栈顶元素。
思路：
首先对于 $b$ 中元素，在一个 $b [i]$ 不为0的元素之前的 $b$ 一定是 $1 ， 2 ， . . . ， b [i]$ 的子序列，所以我们干脆构造出一个合法的 $b$ 数组，对于这个数组，如果某处无值，那么我们就让他等于前一个数+1。

在构造完之后，我们还要判断该数组是否合法，显然，一个数大于他前一个数+1，则说明不合法，而且 $b [1]$ 肯定为1.

在检查之后，我们将整个 $b$ 数组从后往前遍历，我们将1…2…3…4依次放入单调栈中，再根据我们构造的数组b进行弹出即可。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<cstdio>
#include<cmath>
#include<iomanip>
#include<queue>
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
typedef pair<int, int> pii;
const int N = 1e6+10;
int b[N];
int a[N];
int st[N];
int main() {
	int n, k;
	cin >> n >> k;
	for (int i = 1; i <= k; i++) {
		int p;
		cin >> p ;
		cin >> b[p];
	}
	for (int i = 1; i <= n; i++) {
		if (b[i] == 0)b[i] = b[i - 1] + 1;
	}
	int flag = 0;
	for (int i = 1; i < n; i++) {
		if (b[i + 1] - b[i] > 1)flag = 1;
	}
	if (flag) {
		puts("-1");
	}
	else {
		int tt = 0, hh = 0;
		for (int i = n; i >= 1; i--) {
			while(b[i] > tt)st[++tt] = ++hh;
			a[i] = st[tt--];
		}
		cout << a[1];
		for (int i = 2; i <= n; i++)cout << ' ' << a[i];
		cout << endl;
	}
}