poj 2777 (선분 수 + 구간 염색)

2561 단어 데이터 구조
문제 풀이 방향: 이 문 제 는 선분 트 리 + 비트 연산 의 사상 을 이용 했다. 색채 의 종류 가 30 가지 밖 에 없 기 때문에 int 는 저장 할 수 있다. 그래서 우 리 는 선분 트 리 의 노드 에 status 의 상태 정 보 를 추가 하여 이 구간 안의 색채 정 보 를 나타 낸다. 그리고 우 리 는 아버지 노드 의 status 는 두 아들 노드 의 status 가 다 르 거나 얻 은 것 임 을 알 수 있다.또한 검색 효율 을 높이 기 위해 정보 cover 를 설정 해 야 합 니 다. cover = 1 은 이 구간 에 한 가지 색 만 있 음 을 나타 내 고 cover = 0 은 여러 가지 색 이 덮 여 있 음 을 나타 내 므 로 cover = 1 의 노드 를 찾 을 때 직접 거 슬러 올 라 갑 니 다 (하위 노드 는 모두 하나의 색 이기 때문에 더 이상 찾 을 필요 가 없습니다).
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int maxn = 100005;
struct Segment
{
	int l,r;
	int status;
	int cover;
}tree[maxn<<2];
int n,color,m;

void build(int rt,int l,int r)
{
	tree[rt].l = l, tree[rt].r = r;
	tree[rt].status = tree[rt].cover = 1;
	if(l == r) return;
	int mid = (l + r) >> 1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
}

void update(int rt,int l,int r,int val)
{
	if(l <= tree[rt].l && tree[rt].r <= r)
	{
		tree[rt].status = 1 << (val - 1);
		tree[rt].cover = 1;
		return;
	}
	if(tree[rt].cover)
	{
		tree[rt<<1].cover = tree[rt<<1|1].cover = tree[rt].cover;
		tree[rt<<1].status = tree[rt<<1|1].status = tree[rt].status;
		tree[rt].cover = 0;
	}
	int mid = (tree[rt].l + tree[rt].r) >> 1;
	if(mid >= l) update(rt<<1,l,r,val);
	if(mid < r) update(rt<<1|1,l,r,val);
	tree[rt].status = tree[rt<<1].status | tree[rt<<1|1].status;
	if(tree[rt<<1].cover && tree[rt<<1|1].cover && tree[rt<<1].status == tree[rt<<1|1].status)
		tree[rt].cover = 1;
}

int query(int rt,int l,int r)
{
	if(l <= tree[rt].l && tree[rt].r <= r)
		return tree[rt].status;
	if(tree[rt].cover) return tree[rt].status;
	int mid = (tree[rt].l + tree[rt].r) >> 1;
	int ans = 0;
	if(l <= mid) ans |= query(rt<<1,l,r);
	if(mid < r) ans |= query(rt<<1|1,l,r);
	return ans;
}

int main()
{
	char ch;
	int a,b,c;
	while(scanf("%d%d%d",&n,&color,&m)!=EOF)
	{
		build(1,1,n);
		while(m--)
		{
			getchar();
			scanf("%c",&ch);
			if(ch == 'C')
			{
				scanf("%d%d%d",&a,&b,&c);
				if(a > b) swap(a,b);
				update(1,a,b,c);
			}
			else
			{
				scanf("%d%d",&a,&b);
				if(a > b) swap(a,b);
				int ans = query(1,a,b),sum = 0;
				for(int i = 0; i < color; i++)
					if(ans & (1 << i))
						sum++;
				printf("%d
",sum); } } } return 0; }

좋은 웹페이지 즐겨찾기