8.数列分块入门 8

😊 | Powered By HeartFireY | 分块例题

​​#6284. 数列分块入门 8 - 题目 - LibreOJ (loj.ac)​​

给出一个长为$n$的数列，以及$n$个操作，操作涉及区间询问等于一个数$c$的元素，并将这个区间的所有元素改为$c$。

思路很简单，仍然采取分块标记的方式进行记录。

对于区间修改和查询：

• 对于不完整的块，我们先将历史标记下放，执行一次区间修改，然后再暴力查询+修改；
• 对于完整块，如果当前块历史无标记，则暴力遍历统计，然后进行区间修改；如果以已经存在历史标记，则累加块长度至答案中。

#include <bits/stdc++.h>
using namespace std;

const int N = 5e5 + 10;
int id[N], blo;
int a[N], tag[N];

inline void push_down(int pos){
    if(tag[id[pos]] == -1) return;
    for(int i = (id[pos] - 1) * blo + 1; i <= id[pos] * blo; i++) a[i] = tag[id[pos]];
    tag[id[pos]] = -1;
}

int update(int l, int r, int c, int ans = 0){
    push_down(l);
    for(int i = l; i <= min(id[l] * blo, r); i++){
        if(a[i] != c) a[i] = c;
        else ans++;
    }
    if(id[l] == id[r]) return ans;
    push_down(r);
    for(int i = (id[r] - 1) * blo + 1; i <= r; i++){
        if(a[i] != c) a[i] = c;
        else ans++;
    }
    for(int i = id[l] + 1; i <= id[r] - 1; i++){
        if(tag[i] != -1) (tag[i] == c) ? (ans += blo) : (tag[i] = c);
        else{
            for(int j = (i - 1) * blo + 1; j <= i * blo; j++)
                (a[j] == c) ? (ans++) : (a[j] = c);
            tag[i] = c;
        }
    }
    return ans;
}

signed main(){
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    int n = 0; cin >> n; blo = sqrt(n);
    for(int i = 1; i <= n; i++) cin >> a[i];
    for(int i = 1; i <= n; i++) id[i] = (i - 1) / blo + 1, tag[i] = -1;
    for(int i = 1; i <= n; i++){
        int l, r, c; cin >> l >> r >> c;
        cout << update(l, r, c) << endl;
    }
    return 0;
}