多校训练2 K Stack 模拟

题目分析
题目大意：将1-n元素以某个排序方式入栈，若栈顶元素大于要入栈的元素，则将栈顶元素弹出，有一个数组b[ ]，在每次入栈完成后都会有b[ i ] 记录当前栈的元素个数，题目给出元素个数 n 和部分 b[ i ]试构造满足条件的入栈顺序并输出(注意：不止一种可能de入栈顺序,可自行构造)。

思路：
①对于b[ ]数组我们可以确定在最右出现的 b[ i ] = 1处 元素1入栈，同理次右或最右b[ i ] = 2处元素2入栈。且易得出合法的b[ ] 数组，它一定满足b[ i ] - b[ i - 1 ] <= 1(每次入栈操作至多为栈增加一个元素)。
②我们可以根据题目给出的部分b[ ] 数组进行构造，将其补充完整，对于给出的确定的 b[ i ]，我们将1,2……,b[i] - 1从右至左依次赋给 i-1 ~ 前面一个可确定的b[i-1]为止，但要注意可能出现0甚至负数的情况，所以在此我们添加一个max函数，来保证我们填充的b[ ] 最少为1。

accode:

#include <bits/stdc++.h>
using namespace std;
#define max(a, b) (a > b) ? a : b

int main()
{
    std::ios::sync_with_stdio(false); 
    cin.tie(0), cout.tie(0);
    int n, k;
    scanf("%d %d", &n, &k);
    vector<int> b(n);
    vector<bool> vis(n);
    int ad, v;
    for(int i = 0; i < k; i++)
    {
        scanf("%d %d", &ad, &v);
        ad--;
        b[ad] = v;
        vis[ad] = true;
    }
    for(int i = 0; i < n; i++)
    {
        if(vis[i])
        {
            for(int j = i - 1; j >= 0 and not vis[j]; j--)//补充并构造符合条件的b数组
            //将1,2……b[i]-1从右到左依次放入b数组中，直至遇上前一个可确定的b[i-1](即vis[i-1] = true) 
            //要注意可能出现 0甚至负数的情况，所以加上一个max函数 
            {
                b[j] = max(1, b[j + 1] - 1);
            }
        }
    }
    
    for(int i = 0; i < n; i++)//考虑题目给出的部分b[i]里可能没给出b[n-1],导致b[k]右部b数组空白，补充b[k]右部如下 
    {
        if(!b[i])
            b[i] = b[i-1];
    }
    
    bool ok = b[0] = 1;
    for(int i = 1; i < n; i++)
    {
        if(b[i] - b[i-1] > 1)
        {
            ok = false;
            break;
        }
    }
    
    if(not ok)
    {
        cout<<"-1"<<endl;
        return 0;//程序结束 
    }
    
    vector<int> p(n);
    iota(p.begin(), p.end(), 0);//初始化
    sort(p.begin(), p.end(), [&](int x, int y){//确定排放优先顺序序列 
        return b[x] != b[y] ? b[x] < b[y] : x > y;
    });
    //如 b数组为                              1 2 3 3 4
    //即 p里存放的优先顺序为                   1 2 4 3 5
    //后续 a 数组会根据p从小到大顺序填充进 1-n  1 2 4 3 5
    vector<int> a(n);
    for(int i = 0; i < n; i++)
    {
        a[p[i]] = i+1;
    }
    
    for(int i = 0; i < n; i++)
        cout<<a[i]<<" \n"[i == n-1];
    //cout<<endl; 
    
    return 0;
}