首页 > 代码库 > HLG 1808 绘画 (哈希 + map + 线段树)

HLG 1808 绘画 (哈希 + map + 线段树)

链接: 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;
}