[BZOJ2120]数颜色

试题描述

墨墨购买了一套N支彩色画笔(其中有些颜色可能相同),摆成一排,你需要回答墨墨的提问。墨墨会像你发布如下指令: 1、 Q L R代表询问你从第L支画笔到第R支画笔中共有几种不同颜色的画笔。 2、 R P Col 把第P支画笔替换为颜色Col。为了满足墨墨的要求,你知道你需要干什么了吗?

输入

第1行两个整数N,M,分别代表初始画笔的数量以及墨墨会做的事情的个数。第2行N个整数,分别代表初始画笔排中第i支画笔的颜色。第3行到第2+M行,每行分别代表墨墨会做的一件事情,格式见题干部分。

输出

对于每一个Query的询问,你需要在对应的行中给出一个数字,代表第L支画笔到第R支画笔中共有几种不同颜色的画笔。

输入示例


Q
Q
R
Q
Q

输出示例


数据规模及约定

对于100%的数据,N≤10000,M≤10000,修改操作不多于1000次,所有的输入数据中出现的所有整数均大于等于1且不超过10^6。

题解

对于每一个位置 i 我们维护 pre[i] 表示上一个最近的同色的位置。然后用主席树维护 pre[i];需要支持修改,所以得用树状数组套主席树。

对于修改相当于链表的插入和删除,我们对于每一种颜色用一个 set 维护位置集合,修改时需要查前驱后继。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <set>
using namespace std; int read() {
int x = 0, f = 1; char c = getchar();
while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
return x * f;
} #define maxn 10010
#define maxcol 1000010
#define maxnode 7840010 int ToT, sumv[maxnode], lc[maxnode], rc[maxnode];
void update(int& y, int x, int l, int r, int p, int v) {
sumv[y = ++ToT] = sumv[x] + v;
if(l == r) return ;
int mid = l + r >> 1; lc[y] = lc[x]; rc[y] = rc[x];
if(p <= mid) update(lc[y], lc[x], l, mid, p, v);
else update(rc[y], rc[x], mid + 1, r, p, v);
return ;
}
int query(int o, int l, int r, int qr) {
if(r <= qr) return sumv[o];
int mid = l + r >> 1, ans = query(lc[o], l, mid, qr);
if(qr > mid) ans += query(rc[o], mid + 1, r, qr);
return ans;
} int rt[maxn], Rt[maxn];
int n, A[maxn], pre[maxn], lst[maxcol];
void change(int p, int val) {
for(int x = p; x <= n; x += x & -x)
update(Rt[x], Rt[x], 0, n, pre[p], -1),
update(Rt[x], Rt[x], 0, n, val, 1);
pre[p] = val;
return ;
}
int Art[maxn], cntA, Drt[maxn], cntD;
void getrt(int A[], int& cnt, int x) {
cnt = 0;
for(; x; x -= x & -x) A[++cnt] = Rt[x];
return ;
}
int ask(int ql, int qr) {
getrt(Art, cntA, qr); Art[++cntA] = rt[qr];
getrt(Drt, cntD, ql - 1); Drt[++cntD] = rt[ql-1];
int sum = 0;
for(int i = 1; i <= cntA; i++) sum += query(Art[i], 0, n, ql - 1);
for(int i = 1; i <= cntD; i++) sum -= query(Drt[i], 0, n, ql - 1);
return sum;
} set <int> colpos[maxcol]; int main() {
n = read(); int q = read();
for(int i = 0; i < maxcol; i++) colpos[i].insert(0);
for(int i = 1; i <= n; i++) {
A[i] = read();
pre[i] = lst[A[i]];
lst[A[i]] = i;
colpos[A[i]].insert(i);
} for(int i = 1; i <= n; i++) update(rt[i], rt[i-1], 0, n, pre[i], 1);
char cmd[5];
while(q--) {
scanf("%s", cmd);
if(cmd[0] == 'Q') {
int l = read(), r = read();
printf("%d\n", ask(l, r));
}
if(cmd[0] == 'R') {
int p = read(), col = read();
if(A[p] == col) continue; // it -> next ttit -> now tit -> pre
set <int> :: iterator it = colpos[A[p]].lower_bound(p), tit = it, ttit;
it++; tit--;
if(it != colpos[A[p]].end()) change(*it, *tit);
colpos[A[p]].erase(p); colpos[col].insert(p);
it = colpos[col].lower_bound(p); ttit = tit = it;
it++; tit--;
if(it != colpos[col].end()) change(*it, *ttit);
change(*ttit, *tit); A[p] = col;
}
} return 0;
}

最新文章

  1. LightHttpd源码分析
  2. sql server 时间小汇
  3. Merkle Tree学习
  4. tostring格式化输出
  5. java创建Date日期时间笔记
  6. 在linux下实现用ffmpeg把YUV420帧保存成图片
  7. mysql SQL_MODE设置
  8. R语言学习笔记:查看函数的R源代码
  9. 命令行创建Android应用,生成签名,对APK包签名并编译运行
  10. Nexus 刷机
  11. 与后台进行连接,mysql模块 第六篇
  12. CentOS7 PostgreSQL 安装
  13. ThinkPHP 实现验证码渲染、校验、点击刷新
  14. 人工智能 tensorflow框架--&gt;简介及安装01
  15. NFS : device is busy
  16. 虚拟机Linux不能上网简单有效的解决办法
  17. POJ 2029 Get Many Persimmon Trees (模板题)【二维树状数组】
  18. jQuery validator plugin之概要
  19. ZigBee相关网站链接
  20. requests sslerror

热门文章

  1. input标签属性
  2. spring mvc 通过拦截器记录请求数据和响应数据
  3. python工具之exccel模板生成报表
  4. 牛人cad二次开发网站(.net)
  5. 【HEVC简介】DB-DeBlock Filter
  6. 洛谷 P1802 5倍经验日
  7. varchar2(100 char)是什么意思
  8. JavaScript-基础类型和运算符
  9. 关于Ubuntu 16.04中E: Could not get lock /var/lib/dpkg/lock - open的三种解决方案
  10. 屏幕卫士模式系统APP开发