[hdu6183][Color it]

Posted 2021-01-25 wxyww

题目链接

题目大意

有一个矩阵，总共有4种操作

0:清空这个矩阵

1 x y c:将((x,y)(1 leq x ,yleq 10^6))这个点加上一种颜色c((0leq c leq 50))(注意是加上，也就是之前的颜色不会被覆盖)

2 x y1 y2:查询左上角为((1,y1))，右下角为((x,y2))的矩阵中的颜色个数。

3:结束程序

0操作不超过10次，1操作和2操作不超过150000次

思路

先考虑空间足够的情况下应该怎么做。对于每个颜色开一个线段树。第i个位置储存，这个颜色中第i列最小的x是多少，因为每次查询都是从(1,y1)到(x,y2)，所以查询一种颜色的时候，只要看从y1到y2中最小的值是不是比x小就行了，如果比x小，就说明存在这种颜色。

上面的空间复杂度为(50*10^6)，空间不够，所以考虑动态开点线段树。记录下每种颜色的根，然后分别动态开点插入即可。空间复杂度((nlogn)　n=150000)

直接正常查询对于这道题是会超时的，所以要优化一下。也就是如果当前找到这种颜色了，就退出就行了。

代码

/*
* @Author: wxyww
* @Date:   2018-12-10 10:39:02
* @Last Modified time: 2018-12-10 11:20:39
*/
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<ctime>
#include<bitset>
using namespace std;
typedef long long ll;
const int N = 200000 + 100,INF = 1e7,M = 1e6;
ll read() {
    ll x=0,f=1;char c=getchar();
    while(c<'0'||c>'9') {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9') {
        x=x*10+c-'0';
        c=getchar();
    }
    return x*f;
}
int tree[N * 20],ls[N * 20],rs[N * 20],root[55];
int num;
void update(int &rt,int l,int r,int pos,int c) {
    if(!rt) {
        rt = ++num;
        tree[rt] = c;
    }
    tree[rt] = min(tree[rt],c);
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(pos <= mid) update(ls[rt],l,mid,pos,c);
    else update(rs[rt],mid + 1,r,pos,c);
}
int tag,cx;
void query(int rt,int l,int r,int L,int R) {
    if(!rt || tag) return;
    if(L <= l && R >= r) {
        if(tree[rt] <= cx) tag = 1;
        return;
    }
    int mid = (l + r) >> 1;
    if(L <= mid) query(ls[rt],l,mid,L,R);
    if(R > mid)  query(rs[rt],mid + 1,r,L,R); 
}
int main() {
    while(1) {
        int opt = read();
        if(opt == 3) return 0;
        if(opt == 0) {
            memset(tree,0x3f,sizeof(tree));
            memset(root,0,sizeof(root));
            num = 0;
            memset(ls,0,sizeof(ls));
            memset(rs,0,sizeof(rs));
        }
        if(opt == 1) {
            int x = read(),y = read(),c = read();
            update(root[c],1,M,y,x);
        }
        if(opt == 2) {
            cx = read();int y1 = read(),y2 = read();
            int ans = 0;
            for(int i = 0;i <= 50;++i) {
                tag = 0;
                query(root[i],1,M,y1,y2);
                ans += tag;
            }
            printf("%d
",ans);
        }
    }

    return 0;
}