2021牛客暑期多校训练营2 K-Stack

最新推荐文章于 2022-08-03 14:37:37 发布

H A N .

最新推荐文章于 2022-08-03 14:37:37 发布

阅读量89

点赞数

分类专栏： HAN的上古题解

本文链接：https://blog.csdn.net/weixin_49636856/article/details/118934756

版权

HAN的上古题解专栏收录该内容

20 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种基于栈的算法，用于根据给定的b数组重建a数组。题目中给出了伪代码，并通过倒推的方法从b数组恢复a数组。在过程中，如果b数组存在不合理情况（最大值超过数组长度），则输出-1。文章详细解释了如何填充栈以满足b[i]的条件，并提供了完整的C++实现代码。

摘要由CSDN通过智能技术生成

传送门

题目大意懒得写

题目所给伪代码

Stk is an empty stack
for i = 1 to n :
    while ( Stk is not empty ) and ( Stk's top > a[i] ) : 
        pop Stk
    push a[i]
    b[i]=Stk's size

思路

倒退
由B数组推A数组

当栈中的元素数量 n $b_i$ 时，对栈中的数字进行补充。

存在多种情况，可以将当前栈顶元素放在：
1、上一步放置的位置之后
2、上一步放置位置之前 $b_i$ 为 $1$ 处（上一步放置位置之后已有数字）
同时， $b$ 数组也存在多种情况，比较简单的情况：当 $b_i$ 的值未确定时， $b_i=b_{i-1}+1$

若b数组不合理， $b_i$ 的最大值为 $i$ ，所以当 $p_i<x_i$ 时，不存在合理方案，输出 $- 1$

#include<bits/stdc++.h>
using namespace std;
int n,k,x,y,p=0;
int b[1000010],a[1000010];
int main()
{
	int i,j;
	scanf("%d%d",&n,&k);
	for(i=1;i<=k;i++)
	{
		scanf("%d%d",&x,&y);
		b[x]=y;
		if(x<y)
		{
			printf("-1");
			return 0;
		}
	}
	for(i=1;i<=n;i++)
	if(b[i]==0)
	b[i]=b[i-1]+1;
	stack<int> st;
	for(i=n;i>=1;i--)
	{
		if(st.size()<b[i])
			for(j=st.size()+1;j<=b[i];j++)
			{
				p++;
				st.push(p);
			}
		a[i]=st.top();
		st.pop();
	}
	for(i=1;i<=n;i++)
	printf("%d ",a[i]);
	return 0;
}