【Unsolved】[HDOJ4391]Paint The Wall（分块，线段树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4391

题意：初始n个点有不同颜色，两种操作：1：区间染色，2：问区间内某个颜色的出现次数。

线段树维护点，在此基础上分块，发现应该是卡了线段树的log了。在想办法优化掉。

TLE代码：

#include <bits/stdc++.h>
using namespace std;

#define lrt rt << 1
#define rrt rt << 1 | 1
const int maxn = 100100;
typedef struct Seg {
    int v[maxn<<4];
    Seg() { memset(v, -1, sizeof(v)); }

    void pushdown(int rt) {
        if(v[rt] == -1) return;
        v[lrt] = v[rrt] = v[rt];
        v[rt] = -1;
    }

    int query(int pos, int l, int r, int rt) {
        if(l == r) return v[rt];
        pushdown(rt);
        int mid = (l + r) >> 1;
        if(pos <= mid) return query(pos, l, mid, lrt);
        else return query(pos, mid+1, r, rrt);
    }

    void update(int L, int R, int c, int l, int r, int rt) {
        if(L <= l && r <= R) {
            v[rt] = c;
            return;
        }
        pushdown(rt);
        int mid = (l + r) >> 1;
        if(L <= mid) update(L, R, c, l, mid, lrt);
        if(mid < R) update(L, R, c, mid+1, r, rrt);
    }
}Seg;

int n, q, sz, be[maxn];
map<int, int> block[maxn];
Seg seg;

int query(int l, int r, int v) {
    map<int, int> tmp;
    int ret = 0;
    for(int i = l; i <= r; ) {
        if(i % sz == 0 && i + sz <= r) {
            ret += block[be[i]][v];
            i += sz;
        }
        else {
            ret += (v == seg.query(i+1, 1, n, 1));
            i++;
        }
    }
    return ret;
}

void update(int l, int r, int v) {
    for(int i = l; i <= r; ) {
        if(i % sz == 0 && i + sz <= r) {
            block[be[i]].clear(); block[be[i]][v] = sz;
            seg.update(i+1, i+sz, v, 1, n, 1);
            i += sz;
        }
        else {
            int pre = seg.query(i+1, 1, n, 1);
            seg.update(i+1, i+1, v, 1, n, 1);
            block[be[i]][pre]--; block[be[i]][v]++;
            i++;
        }
    }
}

int main() {
    // freopen("in", "r", stdin);
    int cmd, l, r, z;
    while(~scanf("%d%d",&n, &q)) {
        memset(seg.v, 0, sizeof(seg.v));
        for(int i = 0; i <= n; i++) block[i].clear();
        sz = sqrt(n);
        for(int i = 0; i < n; i++) {
            scanf("%d", &z);
            be[i] = i / sz;
            block[be[i]][z]++;
            seg.update(i+1, i+1, z, 1, n, 1);
        }
        while(q--) {
            scanf("%d%d%d%d",&cmd,&l,&r,&z);
            if(cmd == 1) update(l, r, z);
            else printf("%d
", query(l, r, z));
        }
    }
    return 0;
}