小D的灯泡 [根号分治]

$$小D的灯泡$$

---

$正解部分$

将 $N$ 个点建成一个图, 第 $i$ 个节点代表第 $i$ 个位置,

则答案为 亮着的灯泡数量 减去 相邻两个灯泡都是亮着的数量 .

使用 根号分治 求出 后者,

将不同颜色的点分为两类,

1. 个数 小于等于 $\sqrt{N}$ .
2. 个数 大于 $\sqrt{N}$ .

• 修改类型 $1$, 由于个数不超过 $\sqrt{N}$, 对每个节点相邻的 类型 $1$ 节点 正常计算影响, 对 类型 $2$ 节点 $x$ 的影响记到 $num[x]$ 中, 表示与 $x$ 相邻的 类型 $1$ 节点亮灯 的数目, 方便在修改单种类型 $2$ 时快速计算影响 .
• 修改类型 $2$, 类型 $1$ 与 类型 $2$ 相连的情况已经可以由 $num[]$ 解决了,
  考虑 类型 $2$ 与 类型 $2$ 之间的影响怎么计算, 由于个数超过 $\sqrt{N}$ 的灯泡不超过 $\sqrt{N}$ 个,
  考虑建立一个矩阵, 第 $i$ 行 第 $j$ 列表示 类型 $2$ 第 $i$ 个灯泡 与 第 $j$ 个灯泡 之间的相邻的次数, 修改 $A$ 时扫所有与 $A$ 相连的 类型 $2$ 灯泡, 计算贡献即可 .

---

$实现部分$

• 考虑到对 类型$1$ 进行修改的常数较大, 于是可以适当地将阈值设小 .

#include<bits/stdc++.h>
#define reg register
#define pb push_back

int read(){
        char c;
        int s = 0, flag = 1;
        while((c=getchar()) && !isdigit(c))
                if(c == '-'){ flag = -1, c = getchar(); break ; }
        while(isdigit(c)) s = s*10 + c-'0', c = getchar();
        return s * flag;
}

const int maxn = 200005;

int N;
int K;
int Q_;
int n1;
int n2;
int Lim;
int b_cnt;
int B[maxn];
int Mp[maxn];
int is[maxn];
int col[maxn];
int cnt[maxn];
int num[maxn];
int con[3004][3004];

std::vector <int> pos[maxn];

void calc(int x, int y){
        if(is[x] && is[y]) n2 ++;
        if(!is[x] && is[y]) n2 --;
        if(Mp[y]){
                if(is[x]) num[y] ++;
                else num[y] --;
        }
}

void fuck_1(int x){
        is[x] ^= 1;
        if(is[x]) n1 += cnt[x];
        else n1 -= cnt[x];
        for(reg int i = pos[x].size()-1; i >= 0; i --){
                int c = pos[x][i];
                if(c < N) calc(col[c], col[c+1]);
                if(c > 1) calc(col[c], col[c-1]);
        }
}

void fuck_2(int x){
        is[x] ^= 1;
        if(is[x]) n1 += cnt[x], n2 += num[x];
        else n1 -= cnt[x], n2 -= num[x];
        for(reg int i = 1; i <= b_cnt; i ++){
                int cur = B[i];
                if(cur == x || !is[cur]) continue ;
                if(is[x]) n2 += con[Mp[x]][i];
                else if(!is[x]) n2 -= con[Mp[x]][i];
        }
}

void write(int x){
        if(x > 9) write(x/10);
        putchar(x%10 + '0');
}

int main(){
        N = read(), Q_ = read(), K = read();
        for(reg int i = 1; i <= N; i ++) scanf("%d", &col[i]);
        int tmp = 0;
        for(reg int i = 1; i <= N; i ++){
                while(i < N && col[i] == col[i+1]) i ++;
                col[++ tmp] = col[i];
        }
        N = tmp; int Lim = 100;
        for(reg int i = 1; i <= N; i ++) cnt[col[i]] ++, pos[col[i]].pb(i);
        for(reg int i = 1; i <= K; i ++)
                if(cnt[i] > Lim) B[++ b_cnt] = i, Mp[i] = b_cnt;
        for(reg int i = 1; i < N; i ++)
                if(Mp[col[i]] && Mp[col[i+1]]) con[Mp[col[i]]][Mp[col[i+1]]] ++, con[Mp[col[i+1]]][Mp[col[i]]] ++;
        while(Q_ --){
                int x;
                scanf("%d", &x);
                if(!Mp[x]) fuck_1(x);
                else fuck_2(x);
                printf("%d\n", n1 - n2);
        }
        return 0;
}