【bzoj2453】维护队列/【bzoj2120】数颜色 分块+二分
题目描述
你小时候玩过弹珠吗？
小朋友A有一些弹珠，A喜欢把它们排成队列，从左到右编号为1到N。为了整个队列鲜艳美观，小朋友想知道某一段连续弹珠中，不同颜色的弹珠有多少。当然，A有时候会依据个人喜好，替换队列中某个弹珠的颜色。但是A还没有学过编程，且觉得头脑风暴太浪费脑力了，所以向你来寻求帮助。
输入
输入文件第一行包含两个整数N和M。
第二行N个整数，表示初始队列中弹珠的颜色。
接下来M行，每行的形式为“Q L R”或“R x c”，“Q L R”表示A想知道从队列第L个弹珠到第R个弹珠中，一共有多少不同颜色的弹珠，“R x c”表示A把x位置上的弹珠换成了c颜色。
输出
对于每个Q操作，输出一行表示询问结果。
样例输入
2 3
1 2
Q 1 2
R 1 2
Q 1 2
样例输出
2
1
题解
分块+二分
两道一样的题。。。
对于每个i记录一个p[i]，代表i前一个颜色与i相同的位置，没有则为-1。
那么对于区间[l,r]中的i，如果p[i]<l，那么i就是[l,r]中第一个出现c[i]颜色的，可以记录到答案中。
然后分成√n 个块，建立新数组对块中p[i]排序。
查询时，整块使用排序后的数组二分查找小于l的个数，多余部分暴力查询。
修改时，需要考虑3部分：i后面第一个c[j]==c[x]，i后面第一个c[j]==y，i前面第一个c[j]==y。
于是我们还需要建立新数组对块中c[i]排序，并使用二分查找来完成这三个操作。
具体方法：x块内暴力查找，x块外二分查找。
这里压了大量的行，并使用flag减小花括号，不然太多行实在要看吐了QAQ。
还是别忘特判，另外注意i的含义，位置号或块号。
看到网上还有暴力修改的方法，不过亲测很慢，不推荐。
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
int n , si , c[10010] , vc[10010] , p[10010] , vp[10010] , last[1000010];
char str[5];
void reset(int b)
{
int l = b * si , r = min((b + 1) * si - 1 , n - 1) , i;
for(i = l ; i <= r ; i ++ ) vc[i] = c[i] , vp[i] = p[i];
sort(vc + l , vc + r + 1) , sort(vp + l , vp + r + 1);
}
int query(int b , int lim)
{
int l = b * si , r = min((b + 1) * si - 1 , n - 1) , mid , ans = l - 1 , tl = l;
while(l <= r)
{
mid = (l + r) >> 1;
if(vp[mid] < lim) ans = mid , l = mid + 1;
else r = mid - 1;
}
return ans - tl + 1;
}
bool findc(int l , int r , int x)
{
int mid;
while(l <= r)
{
mid = (l + r) >> 1;
if(vc[mid] == x) return 1;
else if(vc[mid] < x) l = mid + 1;
else r = mid - 1;
}
return 0;
}
int main()
{
int m , i , j , x , y , ans , flag , l , r;
scanf("%d%d" , &n , &m);
si = (int)sqrt(n);
memset(last , -1 , sizeof(last));
for(i = 0 ; i < n ; i ++ ) scanf("%d" , &c[i]) , p[i] = last[c[i]] , last[c[i]] = i;
for(i = 0 ; i * si < n ; i ++ ) reset(i);
while(m -- )
{
scanf("%s%d%d" , str , &x , &y);
x -- ;
if(str[0] == 'Q')
{
y -- ;
ans = 0;
if(x / si == y / si) for(i = x ; i <= y ; i ++ ) ans += (p[i] < x);
else
{
for(i = x / si + 1 ; i < y / si ; i ++ ) ans += query(i , x);
for(i = x ; i < (x / si + 1) * si ; i ++ ) ans += (p[i] < x);
for(i = y / si * si ; i <= y ; i ++ ) ans += (p[i] < x);
}
printf("%d\n" , ans);
}
else
{
if(c[x] == y) continue;
flag = 0;
for(i = x + 1 ; !flag && i < (x / si + 1) * si && i < n ; i ++ ) if(p[i] == x) p[i] = p[x] , reset(i / si) , flag = 1;
for(i = x / si + 1 ; !flag && i * si < n ; i ++ )
{
l = i * si , r = min((i + 1) * si - 1 , n - 1);
if(findc(l , r , c[x])) for(j = l ; !flag && j <= r ; j ++ ) if(c[j] == c[x]) p[j] = p[x] , reset(i) , flag = 1;
}
c[x] = y;
flag = 0;
for(i = x + 1 ; !flag && i < (x / si + 1) * si && i < n ; i ++ ) if(c[i] == c[x]) p[i] = x , reset(i / si) , flag = 1;
for(i = x / si + 1 ; !flag && i * si < n ; i ++ )
{
l = i * si , r = min((i + 1) * si - 1 , n - 1);
if(findc(l , r , c[x])) for(j = l ; !flag && j <= r ; j ++ ) if(c[j] == c[x]) p[j] = x , reset(i) , flag = 1;
}
flag = 0;
for(i = x - 1 ; !flag && i >= x / si * si ; i -- ) if(c[i] == c[x]) p[x] = i , flag = 1;
for(i = x / si - 1 ; !flag && ~i ; i -- )
{
l = i * si , r = min((i + 1) * si - 1 , n - 1);
if(findc(l , r , c[x])) for(j = r ; !flag && j >= l ; j -- ) if(c[j] == c[x]) p[x] = j , flag = 1;
}
if(!flag) p[x] = -1;
reset(x / si);
}
}
}
