HDU-6183 Color it

最新推荐文章于 2022-01-02 19:02:52 发布

_zlWang

最新推荐文章于 2022-01-02 19:02:52 发布

阅读量259

点赞数 1

本文链接：https://blog.csdn.net/wl16wzl/article/details/81052427

版权

关于区间查询的问题，想到用线段树来做

但需要先转化一下，每种颜色都建一个线段树，保存Y轴上的区间内最小的X的值

最后查询时，统计每种颜色区间内最小的X是否在范围内就可以了

这样建树需要空间为1e6*4*50，会爆内存，还需要再用动态线段树

动态线段树就是原有线段树基础上，左右子节点的下标不固定了，而是按需求分配保存在l,r数组里

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=1e6+10; //数组大小

int num,root[51],lrt[maxn],rrt[maxn],sum[maxn];
bool flag;
int v;

void init()
{
	num=0;
	memset(root,0,sizeof(root));
	memset(lrt,0,sizeof(lrt));
	memset(rrt,0,sizeof(rrt));
	memset(sum,1e6+10,sizeof(sum));
}

void PushUp(int rt) //向上更新父节点
{
    sum[rt]=min(sum[lrt[rt]],sum[rrt[rt]]);
}

void update(int L,int R,int c,int l,int r,int &rt)
{
	if(!rt) rt=++num;
    if(l==r)
    {
        sum[rt]=min(sum[rt],c);
        return;
    }
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,l,m,lrt[rt]);
    if(m<R) update(L,R,c,m+1,r,rrt[rt]);
    PushUp(rt);
}

void query(int L,int R,int l,int r,int &rt) //查询L~R
{
	if(!rt||flag) return;
    if(L<=l&&R>=r)
    {
    	if(sum[rt]<=v) flag=true;
    	return;
    }
    int m=(l+r)>>1;
    if(L<=m) query(L,R,l,m,lrt[rt]);
    if(m<R) query(L,R,m+1,r,rrt[rt]);
}

int main()
{
	int op,x,y,c;
	init();
	while(~scanf("%d",&op))
	{
		if(op==3) break;
		if(op==0)
		{
			init();
		}
		else if(op==1)
		{
			scanf("%d%d%d",&x,&y,&c);
			update(y,y,x,1,1e6,root[c]);
		}
		else if(op==2)
		{
			scanf("%d%d%d",&x,&y,&c);
			v=x;
			int ans=0;
			for(int i=0;i<=50;i++)
			{
				flag=false;
				query(y,c,1,1e6,root[i]);
				if(flag) ans++;
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}