小红的陡峭值 牛客周赛Round 34

题目描述

小红定义一个数组的陡峭值为相邻两数差的绝对值之和。
现在小红拿到了一个长度为n的、仅由正整数组成的数组，但她有一些元素看不清了，只记得数组的陡峭值恰好等于 1。
小红希望你能还原整个数组，你能帮帮她吗？

输入描述:

第一行输入一个正整数n，代表数组的大小。
第二行输入n个整数aia_iai​，其中如果ai​为 0 代表小红看不清该元素，大于 0 代表能看清该元素。
2≤n≤10^5
0≤ai≤10^9

输出描述:

如果无解，请输出 -1。
否则输出n个正整数，用空格隔开，代表小红还原的数组。
有多解时输出任意即可。

示例1

输入

4
9 0 0 8

输出

9 9 9 8

示例2

输入

3
0 1 0

输出

1 1 2

示例3

输入

5
0 1 2 1 0

输出

-1

---

AC代码： 

#include<iostream>
#include<queue>
#include<cmath>
#include<algorithm>

using namespace std;

typedef long long LL;
const int N = 1e5+10;
int a[N],vis[N];
int n;

int main()
{
    cin.tie(nullptr)->ios::sync_with_stdio(false);
    cin >> n;
    //优先队列（先进先出）
    priority_queue <int> pq;
    for(int i=0;i<n;i++)
    {
        cin >> a[i];
        //把 不为0的坐标全部存入到pq中
        if(a[i])
        {
            pq.push(i);
            vis[i] = 1;//做个标记
        }
    }
    //如果数组元素全部为0构造成211111；
    if(pq.size() == 0)
    {
        cout << 2 << ' ';
        for(int i=1;i<n;i++)
        {
            cout << 1 << ' ';
        }
        return 0;
    }
    //数组中元素是不为0我们每次取出首个不为0元素坐标
    while(pq.size()!=0)
    {
        auto t = pq.top();
        pq.pop();
        //利用两个if使其数组变成aaaaabbbb形式
        if(t > 0 && a[t-1] == 0)
        {
            a[t-1] = a[t];
            pq.push(t-1);
        }
        if(t < n-1 && a[t+1] == 0)
        {
            a[t+1] = a[t];
            pq.push(t+1);
        }
    }
    //求一个ans的值，
    /*三种情况，
    第一种：ans = 0，说明构造完的元素里面都一样，需要判断一下首元素a[0]h和
    尾部元素a[n-1]是否有之前是0 ，有一个是的话就有解，否则无解
    第二种 ans > 1,这种直接pass
    第三种 ans = 1,这种有解
    */
    LL ans = 0;
    for(int i=1;i<n;i++)
    {
        ans = ans + abs(a[i] - a[i-1]);
    }
    
    if(ans > 1)
    {
        cout << -1;
        return 0;
    }
    else if(ans == 0)
    {
        //例如：010这组构造完是111，判断一下
        //a[0] 和 a[n-1] 没构造之前是不是0
        if(vis[0] && vis[n-1])
        {
            cout << -1;
            return 0;
        }
        if(vis[0] == 0)
        {
            a[0]++;
        }
        else
        {
            a[n-1] ++;
        }
        for(int i=0;i<n;i++)
            cout << a[i] << ' ';
    }
    else //ans = 1的情况
    {
        for(int i=0;i<n;i++)
        {
            cout << a[i] << ' ';
        }
    }
    return 0;
}