2021牛客暑期多校训练营2 - K - Stack（ 栈 + 拓扑 ）

题目链接：点击进入

题目

题意

给出 b 数组的求解方法( 根据 a 得到 b )，然后给你若干个 b 数组的值，让你求一个合法的 a 数组

思路

贪心模拟栈运行的过程，从小到大枚举 b 的每个位置。
若是 b [ i ] 这个位置没数，那就入栈一个比栈顶大的数（ 我们这里入栈的是下标，意味着 a [ i ] > a [ 栈顶 ] ） ，让 b [ i ] = b [ i - 1 ] + 1 ；
若是 b [ i ] 这个位置有数，那么我们先判断 b [ i ] 与 栈元素个数的关系：
若是 b [ i ] > 栈内元素个数 + 1 ( 也就是往栈内加一个数达不到 b[ i ] )，说明这个 a 数组不存在，因为我们是贪心的放的，若是这样都满足不了 b [ i ] ，只能说它不合法。
若是 b [ i ] <= 栈内元素个数 + 1 ，那么就让栈内元素出栈 ，一直到 栈内元素数 = b [ i ] - 1 （ 因为还要入栈 a [ i ] ），出栈是为了满足构造 b [ i ]，同时也意味着出栈的元素代表的值都比 a [ i ] 大，同时出栈结束的栈顶代表的值比 a [ i ] 小。
这个枚举过程下来，我们会得到 a 数组内不同下标对应值之间的大小关系，根据这个大小关系我们可以连边进行拓扑排序（ 小的向大的连边，这样拓扑出来的是一个对应值从小到大排列的下标数组 ），对得到的这个下标数组，从前往后对它们对应的值，从小到大赋值即可。

代码

//#pragma GCC optimize(3)//O3
//#pragma GCC optimize(2)//O2
#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
//#define int long long
//#define double long double
//#define rep(i,x,y) for(register int i = x; i <= y;++i)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=998244353;
const double eps=1e-9;
const int N=5e3+10;

//inline int read()
//{
//    int k = 0, f = 1 ;
//    char c = getchar() ;
//    while(!isdigit(c)){if(c == '-') f = -1 ;c = getchar() ;}
//    while(isdigit(c)) k = (k << 1) + (k << 3) + c - 48 ,c = getchar() ;
//    return k * f ;
//

int n,k,b[maxn],x,ans[maxn],cnt;
int deg[maxn];
stack<int>st;
vector<int>v[maxn];

void tuopu()
{
	queue<int>q;
	for(int i=1;i<=n;i++)
		if(!deg[i])
			q.push(i);
	while(q.size())
	{
		int cur=q.front();
		q.pop();
		ans[cur]=++cnt;
		for(auto t:v[cur])
		{
			deg[t]--;
			if(!deg[t])
				q.push(t);
		}
	}
			
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	cin>>n>>k;
	for(int i=1;i<=k;i++)
	{
		cin>>x;
		cin>>b[x];
	}
	int flag=0;
	for(int i=1;i<=n;i++)
	{
		if(b[i])
		{
			if(b[i]>st.size()+1)
			{
				flag=1;
				break;
			}
			while(st.size()&&b[i]<=st.size())
			{
				v[i].push_back(st.top());
				deg[st.top()]++;
				st.pop();
			}
		}
		if(st.size())
		{
			v[st.top()].push_back(i);
			deg[i]++;
		}
		st.push(i);
	}
	tuopu();
	if(flag) cout<<-1<<endl;
	else
		for(int i=1;i<=n;i++)
			cout<<ans[i]<<' ';
    return 0;
}