Codeforces 1023D Array Restoration（线段树）

题目链接：Array Restoration

题意

有一个长度为 $n$ 的序列，可以对这 $n$ 个序列进行 $q$ 次操作，第 $i$ 次操作选择一个区间 $[l_i,r_i]$，将这个区间内的所有数字用 $i$ 替换，每个位置上的数字至少被一个区间覆盖，最后将这个序列中的某几个数字改为 $0$。现在题目给出最终的序列，问是否存在合法的 $q$ 次操作，能够得到给出的序列，求 $q$ 次操作后（将序列某些数字变为 $0$ 之前）的序列。

输入

第一行为两个整数 $n,q~(1\leq n,q\leq2\times10^5)$，第二行为 $n$ 个整数 $a_1,a_2,\cdots,a_n~(0\leq a_i\leq q)$。

输出

如果给出的序列无法从上面的操作中得到，输出 $NO$，否则在第一行输出 $YES$，第二行输出 $q$ 次操作之后的 $n$ 个整数。

样例

输入
4 3 1 0 2 3
输出
YES 1 2 2 3
提示
将 $0$ 用 $1$ 替换也是合法的，但是用 $3$ 替换是不合法的。

输入
3 10 10 10 10
输出
YES 10 10 10
提示
不论前 $q-1$ 次操作是怎么样，第 $q$ 次操作是选择区间 $[1,3]$ 即可。

输入
5 6 6 5 6 2 2
输出
NO
提示
由于第 $5$ 次操作必须在第 $6$ 次操作之前，所以不可能先将区间 $[1,3]$ 更新为 $6$ 之后再将区间 $[2,2]$ 更新为 $5$。

输入
3 5 0 0 0
输出
YES 5 4 2
提示
有许多种操作对于 $0~0~0$ 而言都是合法的。

题解

由于数字大的一定比数字小的后更新，所以每个数字出现的左端点与右端点之间的数字不能小于它自身，可以用线段树维护最小值，从 $q$ 到 $1$ 询问第 $i$ 个数字出现的左端点 $L_i$ 与右端点 $R_i$，如果在查询区间最小值时遇到 $0$ 的情况，就向下将 $0$ 更新为当前数字再进行查询。记得首先查询序列内是否有数字 $q$，如果没有数字 $q$ 就将其中任意一个 $0$ 修改为 $q$，如果无法修改则直接输出 $NO$。所有询问与更新结束后，如果序列中仍含有 $0$，就将所有 $0$ 赋值为 1 1 <script type="math/tex" id="MathJax-Element-45">1</script>。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <functional>
#include <algorithm>
using namespace std;

#define LL long long
const int maxn = 200000 + 100;
int n, q;
int num[maxn], L[maxn], R[maxn];
int Min[maxn << 2];

void push_up(int rt) {
    Min[rt] = min(Min[rt << 1], Min[rt << 1 | 1]);
}

void build(int l, int r, int rt) {
    if(l == r) {
        Min[rt] = num[l];
        return ;
    }
    int m = (l + r) >> 1;
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
    push_up(rt);
}

void update(int x, int l, int r, int rt) {
    if(l == r) {
        Min[rt] = x;
        num[l] = x;
        return ;
    }
    int m = (l + r) >> 1;
    if(Min[rt << 1] == 0) {
        update(x, l, m, rt << 1);
    }
    if(Min[rt << 1 | 1] == 0) {
        update(x, m + 1, r, rt << 1 | 1);
    }
    push_up(rt);
}

int query(int L, int R, int x, int l, int r, int rt) {
    if(L <= l && r <= R) {
        if(Min[rt] == 0) {
            update(x, l, r, rt);
        }
        return Min[rt];
    }
    int m = (l + r) >> 1;
    int ret = INT_MAX;
    if(L <= m) {
        ret = min(ret, query(L, R, x, l, m, rt << 1));
    }
    if(m < R) {
        ret = min(ret, query(L, R, x, m + 1, r, rt << 1 | 1));
    }
    push_up(rt);
    return ret;
}

int main() {
    #ifdef Dmaxiya
    freopen("test.txt", "r", stdin);
//    freopen("10.out", "w", stdout);
    #endif // Dmaxiya

    while(scanf("%d%d", &n, &q) != EOF) {
        for(int i = 0; i <= q; ++i) {
            L[i] = n + 1;
            R[i] = 0;
        }
        for(int i = 1; i <= n; ++i) {
            scanf("%d", &num[i]);
            L[num[i]] = min(L[num[i]], i);
            R[num[i]] = max(R[num[i]], i);
        }
        if(L[q] > R[q]) {
            if(L[0] != n + 1) {
                num[L[0]] = q;
                L[q] = R[q] = L[0];
            } else {
                printf("NO\n");
                continue;
            }
        }
        build(1, n, 1);
        bool flag = true;
        for(int i = q; i >= 1; --i) {
            if(L[i] > R[i]) {
                continue;
            }
            if(query(L[i], R[i], i, 1, n, 1) < i) {
                flag = false;
                break;
            }
        }
        if(!flag) {
            printf("NO\n");
            continue;
        }
        printf("YES\n");
        for(int i = 1; i <= n; ++i) {
            if(i != 1) {
                printf(" ");
            }
            if(num[i] == 0) {
                printf("1");
            } else {
                printf("%d", num[i]);
            }
        }
        printf("\n");
    }

    return 0;
}