zoukankan      html  css  js  c++  java
  • POJ 2777 Count Color(线段树染色,二进制优化)

    Count Color
    Time Limit: 1000MS   Memory Limit: 65536K
    Total Submissions: 42940   Accepted: 13011

    Description

    Chosen Problem Solving and Program design as an optional course, you are required to solve all kinds of problems. Here, we get a new problem. 

    There is a very long board with length L centimeter, L is a positive integer, so we can evenly divide the board into L segments, and they are labeled by 1, 2, ... L from left to right, each is 1 centimeter long. Now we have to color the board - one segment with only one color. We can do following two operations on the board: 

    1. "C A B C" Color the board from segment A to segment B with color C. 
    2. "P A B" Output the number of different colors painted between segment A and segment B (including). 

    In our daily life, we have very few words to describe a color (red, green, blue, yellow…), so you may assume that the total number of different colors T is very small. To make it simple, we express the names of colors as color 1, color 2, ... color T. At the beginning, the board was painted in color 1. Now the rest of problem is left to your. 

    Input

    First line of input contains L (1 <= L <= 100000), T (1 <= T <= 30) and O (1 <= O <= 100000). Here O denotes the number of operations. Following O lines, each contains "C A B C" or "P A B" (here A, B, C are integers, and A may be larger than B) as an operation defined previously.

    Output

    Ouput results of the output operation in order, each line contains a number.

    Sample Input

    2 2 4
    C 1 1 2
    P 1 2
    C 2 2 2
    P 1 2
    

    Sample Output

    2
    1
    

    Source

    题目链接:POJ 2777

    大致题意就是区间染色覆盖,区间查询出现的颜色种类。跟以往的普通线段树倒是有点不一样,就是如何进行颜色的记录问题,看了一下别人的做法,就是用2的幂数来代表颜色,由于题目中颜色小于等于30种,小于log2(INT_MAX),可以存的下,那怎么表示呢?看下面的表格

    1 1
    2 10
    3 100
    4 1000
    5 10000
    6 100000
    .. .....
    .. .....
    .. .....
    30 100000..(29个0)


    由以上表格可以看出,某种颜色用2的幂来表示相当于二进制的1和(颜色标号-1)个0,比如6用2的幂表示就是1和5个0。

    那为什么要用这个表示呢?因为后面还要用到按位或“|”运算,即重复的二进制中的1也只算一个,只要出现过,就能算计去,出现两次,也只会按位或得到一个1,比如一棵树有5(表示为10000)和2(表示为10),按位或后就得到10010,其中的1的个数就是颜色个数,不信再试试,再和颜色2按位或(表示为10),还是10010,这显然是符合同色只算一次的情况的,然后就稍微改写一下向上更新、向下更新函数,就差不多了,计算出最后的query结果中1的个数也就是答案了。用了bitset的简化求二进制1个数的写法挺方便

    代码:

    #include<iostream>
    #include<algorithm>
    #include<cstdlib>
    #include<sstream>
    #include<cstring>
    #include<cstdio>
    #include<bitset>
    #include<string>
    #include<deque>
    #include<stack>
    #include<cmath>
    #include<queue>
    #include<set>
    #include<map>
    using namespace std;
    #define INF 0x3f3f3f3f
    #define CLR(x,y) memset(x,y,sizeof(x))
    #define LC(x) (x<<1)
    #define RC(x) ((x<<1)+1)
    #define MID(x,y) ((x+y)>>1)
    typedef pair<int,int> pii;
    typedef long long LL;
    const double PI=acos(-1.0);
    const int N=100010;
    struct seg
    {
    	int l,mid,r;
    	int color;
    	int add;
    }T[N<<2];
    void pushup(int k)
    {
    	T[k].color=T[LC(k)].color|T[RC(k)].color;
    }
    void pushdown(int k)
    {
    	if(T[k].add)
    	{
    		T[LC(k)].color=T[k].add;
    		T[RC(k)].color=T[k].add;
    		T[LC(k)].add=T[k].add;
    		T[RC(k)].add=T[k].add;
    		T[k].add=0;	
    	}
    }
    void build(int k,int l,int r)
    {
    	T[k].l=l;
    	T[k].r=r;
    	T[k].mid=MID(l,r);
    	T[k].add=0;
    	T[k].color=0;
    	if(l==r)
    	{
    		T[k].color=1;
    		return ;
    	}
    	build(LC(k),l,T[k].mid);
    	build(RC(k),T[k].mid+1,r);
    	pushup(k);
    }
    void update(int k,int l,int r,int color)
    {
    	if(r<T[k].l||l>T[k].r)
    		return ;
    	if(l<=T[k].l&&r>=T[k].r)
    	{
    		int bina_val=1<<(color-1);
    		T[k].color=T[k].add=bina_val;
    		return ;
    	}
    	pushdown(k);
    	if(l<=T[k].mid)
    		update(LC(k),l,r,color);
    	if(r>T[k].mid)
    		update(RC(k),l,r,color);
    	pushup(k);
    }
    int query(int k,int l,int r)
    {
    	if(l<=T[k].l&&T[k].r<=r)
    		return T[k].color;
    	else
    	{
    		pushdown(k);
    		if(r<=T[k].mid)
    			return query(LC(k),l,r);
    		else if(l>T[k].mid)
    			return query(RC(k),l,r);
    		else
    			return query(LC(k),l,T[k].mid)|query(RC(k),T[k].mid+1,r);
    	}
    }
    int main(void)
    {
    	int L,t,O;
    	int i,j,l,r,c;
    	char ops[3];
    	while (~scanf("%d%d%d",&L,&t,&O))
    	{
    		build(1,1,L);
    		for (i=0; i<O; ++i)
    		{
    			scanf("%s",ops);
    			if(ops[0]=='C')
    			{
    				scanf("%d%d%d",&l,&r,&c);
    				if(l>r)
    					swap(l,r);
    				update(1,l,r,c);
    			}
    			else
    			{
    				scanf("%d%d",&l,&r);
    				if(l>r)
    					swap(l,r);
    				printf("%d
    ",bitset<32>(query(1,l,r)).count());
    			}
    		}
    	}
    	return 0;
    }
  • 相关阅读:
    数据库02
    MySQL1
    GIL 死锁 递归锁 event 信号量 线程Queue
    小脚本 暴力删除文件 刷屏
    常见web攻击 及基础 回顾(杂记)
    接口中的简单异步 async
    python协程 示例
    python 利用jinja2模板生成html
    python 调用webservices 接口
    python 进程 进程池 进程间通信
  • 原文地址:https://www.cnblogs.com/Blackops/p/5766267.html
Copyright © 2011-2022 走看看