带修改莫队模板

bzoj2120

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
const int mod = 1e9+9;
typedef long long  ll;
int Case = 1, n, m, B;
int sum[maxn];
int cc[maxn], _hash[maxn];
int res[maxn];
struct node{
    int l, r;
    int id, cur;
    bool operator <(const node x)const {
        if(_hash[l] == _hash[x.l] && r == x.r) return cur < x.cur;
        else if(_hash[l] == _hash[x.l]) return r < x.r;
        else return _hash[l] < _hash[x.l];
    }
}q[maxn];

struct point{
    int x, y;
}ch[maxn];
void update(int &now, int pos, int flag) {
    if(!sum[cc[pos]] && flag == 1) now++;
    if(sum[cc[pos]] == 1 && flag == -1) now--;
    sum[cc[pos]] += flag;
}

void updatet(int &now, int pos, int l, int r) {
    if(ch[pos].x >= l && ch[pos].x <= r) update(now, ch[pos].x, -1);
    swap(cc[ch[pos].x], ch[pos].y);
    if(ch[pos].x >= l && ch[pos].x <= r) update(now, ch[pos].x, 1);
}

void cal(int cu) {
    int l =  1, r = 0, cur = 0;
    int now = 0;
    for(int i = 1; i <= cu; i++) {
        while(l < q[i].l) update(now, l++, -1);
        while(r > q[i].r) update(now, r--, -1);
        while(l > q[i].l) update(now, --l, 1);
        while(r < q[i].r) update(now, ++r, 1);
        while(cur < q[i].cur) updatet(now, ++cur, l, r);
        while(cur > q[i].cur) updatet(now, cur--, l, r);
        res[q[i].id] = now;
    }
}

void solve() {
    cin>>n>>m;
    B = sqrt(n);
    for(int i = 1; i <= n; i++) {
        cin>>cc[i];
        _hash[i] = i/B;
    }
    int cnt = 0, cu = 0;
    for(int i = 1;  i<= m; i++) {
        string op;int x, y;
        cin>>op;
        if(op == "Q") {
            cu++;
            cin>>q[cu].l>>q[cu].r;
            q[cu].cur = cnt;
            q[cu].id = cu;
        }
        else{
            cnt++;
            cin>>ch[cnt].x>>ch[cnt].y;
        }
    }
    sort(q+1, q+1+cu);
    cal(cu);
    for(int i = 1; i <= cu; i++) {
        cout<<res[i]<<"\n";
    }
}
int main() {
    ios::sync_with_stdio(false);
    //freopen("in.txt", "r", stdin);
    while(Case--) {
        solve();
    }
    return 0;
}