HDU 6183 Color it(线段树 动态开点)

题意：有④种操作：①1 x y c表示在(x, y)点染上颜色c；②2 X y1 y2表示查询在(1, y1)到(X, y2)范围内有多少种不同的颜色；③0表示清屏；④3表示程序退出（0<=x, y<=1000000, 0<=c<=50）

思路:因为只有50种颜色，所以我们可以分别考虑每个颜色，x轴每次都是询问的[1,x]，所以我们开50棵线段树，对于

每种颜色维护每个y坐标的最小x，查询的时候就是区间查询[y1,y2]的最小x是否小于当前查询的x。因为50棵1e6的线

段树会炸空间，所以需要动态开点。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
struct node
{
    int l, r, v;
    node() {}
    node(int ll, int rr, int vv):l(ll), r(rr), v(vv) {}
}tree[maxn*4];
int tot, rt[51], have;

void update(int &root, int l, int r, int pos, int val)
{
    if(!root) tree[root=++tot] = node(0, 0, val);
    if(val < tree[root].v) tree[root].v = val;
    if(l == r) return ;
    int mid = (l+r)/2;
    if(pos <= mid) update(tree[root].l, l, mid, pos, val);
    else update(tree[root].r, mid+1, r, pos, val);
}

void query(int root, int l, int r, int i, int j, int x)
{
    if(have || !root) return ;
    if(i <= l && j >= r)
    {
        if(tree[root].v <= x) have = 1;
        return ;
    }
    int mid = (l+r)/2;
    if(i <= mid) query(tree[root].l, l, mid, i, j, x);
    if(j > mid) query(tree[root].r, mid+1, r, i, j, x);
}

int main(void)
{
//    freopen("in.txt", "r", stdin);
    int cmd, x, y, y1, y2, c;
    while(1)
    {
        scanf("%d", &cmd);
        if(cmd == 3) break;
        if(cmd == 0)
        {
            for(int i = 0; i < 51; i++)
                rt[i] = 0;
            tot = 0;
        }
        else if(cmd == 1)
        {
            scanf("%d%d%d", &x, &y, &c);
            update(rt[c], 1, maxn, y, x);
        }
        else
        {
            scanf("%d%d%d", &x, &y1, &y2);
            int ans = 0;
            for(int i = 0; i < 51; i++)
            {
                have = 0;
                query(rt[i], 1, maxn, y1, y2, x);
                ans += have;
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}