链接: http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1808
Description
小胖子最近喜欢上了画画,但是他的画画技术很差,只能从零开始,小胖子在一条线上画了很多彩色的球,但是觉得不好看,就想修改。
小胖子站在画前想了又想,他有时会将一些连续的点涂成相同的颜色。但他感觉累的时候,就会无聊的数数某个颜色在某段区间内出现的次数。
Input
每组数据的第一行输入两个正整数n和m(1<-n,m<=100000),n代表彩球的个数长度,m代表询问次数。
接下来一行包括n个整数代表每个位置的颜色。
接下来m行,每行包括四个正整数a,l,r,z(1<=a<=2, 0<=l<=r<n ,0<=z<2^31)
a=1代表将l到r之间的彩球涂成z;
a=2代表小胖子想知道有多少个颜色为z的彩球在l到r出现了多少次。
Output
输出每次询问的答案。
Sample Input
5 5
1 2 3 4 0
2 1 3 3
1 1 3 1
2 1 3 3
2 0 3 1
2 3 4 1
Sample Output
1
0
4
1
代码及解析如下:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <map>
#include <algorithm>
#define MAXN 100005
#define RST(N)memset(N, 0, sizeof(N))
using namespace std;
int n, m ,bsize, bnum, x[MAXN];
int q, l, r, z;
struct hash_block {
int cls, size;
std::map <int, int> mp;
}b[350];
//下传标记,当这个块化整为零的时候需要下传标记并跟新所有元素
void pushdown(int id)
{
hash_block &hb = b[id];
if(hb.cls != -1) {
for(int i=id*bsize; i<id*bsize+hb.size; i++) x[i] = hb.cls;
hb.mp.clear(), hb.mp[hb.cls] = hb.size;
hb.cls = -1;
}
}
//更新,中间的部分打标记就可以了,两边的sqrt(n)暴力更新
void update(int l, int r, int c)
{
int lb = l/bsize, rb = r/bsize, ans = 0;
for(int i=lb+1; i<rb; i++) b[i].cls = c;
if(lb != rb) {
pushdown(lb), pushdown(rb);
for(int i=l; i<lb*bsize+b[lb].size; i++) {
b[lb].mp[x[i]]--;
b[lb].mp[c]++;
x[i] = c;
}
for(int i=rb*bsize; i<=r; i++) {
b[rb].mp[x[i]]--;
b[rb].mp[c]++;
x[i] = c;
}
}else {
pushdown(lb);
for(int i=l; i<=r; i++) {
b[lb].mp[x[i]]--;
b[lb].mp[c]++;
x[i] = c;
}
}
return ;
}
//中间的部分根据标记或者hash表可以直接查询,两边的sqrt(n)暴力查询
int query(int l, int r, int c)
{
int lb = l/bsize, rb = r/bsize, ans = 0;
for(int i=lb+1; i<rb; i++) {
//一直错在这个地方了,如果有标记,直接判断标记是不是需要的颜色
//如果没标记,要先判map中有没有这个元素然后在操作,否则会MLE!
if(b[i].cls == c) ans += b[i].size;
else if(b[i].cls == -1 && b[i].mp.find(c) != b[i].mp.end()) ans += b[i].mp[c];
}
if(lb != rb) {
pushdown(lb), pushdown(rb);
for(int i=l; i<lb*bsize+b[lb].size; i++) ans += (x[i]==c);
for(int i=rb*bsize; i<=r; i++) ans +=( x[i]==c);
}else {
pushdown(lb);
for(int i=l; i<=r; i++) ans += (x[i]==c);
}
return ans;
}
void initblock()
{
bsize = (int)sqrt(n+1e-8);
bnum = (n-1)/bsize + 1;
for(int i=0; i<bnum; i++) {
b[i].mp.clear();
b[i].cls = -1;
b[i].size = std::min(i*bsize+bsize, n) - i*bsize;
}
for(int i=0; i<n; i++){
scanf("%d", &x[i]);
b[i/bsize].mp[x[i]]++;
}
}
int main()
{
while(~scanf("%d %d", &n, &m)) {
initblock();
while(m--) {
scanf("%d %d %d %d", &q, &l, &r, &z);
if(q == 1) update(l, r, z);
else printf("%d\n", query(l, r, z));
}
}
return 0;
}