题目链接:点击进入
题目
题意
给出 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;
}