小 D 的 灯 泡 小D的灯泡 小D的灯泡
正 解 部 分 \color{red}{正解部分} 正解部分
将 N N N 个点建成一个图, 第 i i i 个节点代表第 i i i 个位置,
则答案为 亮着的灯泡数量 减去 相邻两个灯泡都是亮着的数量 .
使用 根号分治 求出 后者,
将不同颜色的点分为两类,
- 个数 小于等于 N \sqrt{N} N .
- 个数 大于 N \sqrt{N} N .
- 修改类型 1 1 1, 由于个数不超过 N \sqrt{N} N, 对每个节点相邻的 类型 1 1 1 节点 正常计算影响, 对 类型 2 2 2 节点 x x x 的影响记到 n u m [ x ] num[x] num[x] 中, 表示与 x x x 相邻的 类型 1 1 1 节点亮灯 的数目, 方便在修改单种类型 2 2 2 时快速计算影响 .
- 修改类型
2
2
2, 类型
1
1
1 与 类型
2
2
2 相连的情况已经可以由
n
u
m
[
]
num[]
num[] 解决了,
考虑 类型 2 2 2 与 类型 2 2 2 之间的影响怎么计算, 由于个数超过 N \sqrt{N} N 的灯泡不超过 N \sqrt{N} N 个,
考虑建立一个矩阵, 第 i i i 行 第 j j j 列表示 类型 2 2 2 第 i i i 个灯泡 与 第 j j j 个灯泡 之间的相邻的次数, 修改 A A A 时扫所有与 A A A 相连的 类型 2 2 2 灯泡, 计算贡献即可 .
实 现 部 分 \color{red}{实现部分} 实现部分
- 考虑到对 类型 1 1 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;
}