HDU6183 Color it(线段树 动态开点)

来源:互联网 发布:4g卡为什么显示3g网络 编辑:程序博客网 时间:2024/06/05 00:50

题意:有④种操作:①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;}