【线段树（lazy）】poj 2777 Count Color

最新推荐文章于 2022-03-12 11:46:19 发布

原创最新推荐文章于 2022-03-12 11:46:19 发布 · 488 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #线段树

数据结构同时被 2 个专栏收录

38 篇文章

订阅专栏

POJ

31 篇文章

订阅专栏

本文介绍了一种基于区间更新和查询的算法实现，该算法适用于解决特定类型的问题，如将长板按顺序涂色并查询指定区间内的不同颜色数量。通过递归地构建、更新和查询树状数据结构来高效完成任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://poj.org/problem?id=2777

题意：把长度为L的板子分成L份，标签从1~L，开始时板子的颜色都为1,然后对板子的某一区间涂色，求在某一区间内有多少种不同的颜色？（A可能>B）

分析：区间更新+区间查找

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int NM=100005;
int T[NM*4];
bool f[35];

void Build(int lf,int rg,int rt)
{
	T[rt]=1;
	if(lf==rg) return;
	int mid=(lf+rg)>>1;
	Build(lf,mid,rt<<1);
	Build(mid+1,rg,rt<<1|1);
}

void Update(int x,int y,int vau,int lf,int rg,int rt)
{
	if(T[rt]==vau) return;
	if(x==lf&&y==rg){
		T[rt]=vau;
		return;
	}
	if(T[rt]!=-1){
		T[rt<<1]=T[rt<<1|1]=T[rt];
		T[rt]=-1;
	}
	
	int mid=(lf+rg)>>1;
	if(x>mid) Update(x,y,vau,mid+1,rg,rt<<1|1);
	else if(y<=mid) Update(x,y,vau,lf,mid,rt<<1);
	else {
		Update(x,mid,vau,lf,mid,rt<<1);
		Update(mid+1,y,vau,mid+1,rg,rt<<1|1);
	}
}

void Query(int x,int y,int lf,int rg,int rt)
{
	if(T[rt]!=-1)
		f[T[rt]]=1;
	else {
		int mid=(lf+rg)>>1;
		if(x>mid) Query(x,y,mid+1,rg,rt<<1|1);
		else if(y<=mid) Query(x,y,lf,mid,rt<<1);
		else {
			Query(mid+1,y,mid+1,rg,rt<<1|1);
			Query(x,mid,lf,mid,rt<<1);
		}
	}
}

int main()
{
	int i,n,m,t,x,y,vau,ans;
	char str[3];
	while(scanf("%d%d%d",&n,&t,&m)!=EOF)
	{	
		Build(1,n,1);
		while(m--)
		{
			scanf("%s",str);
			if(str[0]=='C'){
				scanf("%d%d%d",&x,&y,&vau);
				if(x>y) swap(x,y);
				Update(x,y,vau,1,n,1);
			}
			else {
				scanf("%d%d",&x,&y);
				memset(f,0,sizeof(f));
				if(x>y) swap(x,y);
				Query(x,y,1,n,1);
				ans=0;
				for(i=1;i<=t;i++)
					if(f[i]) ans++;
				printf("%d\n",ans);
			}
		}	
	}
	return 0;
}