poj2777-----线段树------位运算模拟线段染色------求区间颜色种数

最新推荐文章于 2018-09-01 17:34:41 发布

monkey7777

最新推荐文章于 2018-09-01 17:34:41 发布

阅读量691

点赞数

分类专栏： POJ解题之路

本文链接：https://blog.csdn.net/monkey7777/article/details/9297117

版权

POJ解题之路专栏收录该内容

40 篇文章 0 订阅

订阅专栏

树状数组建立线段树，每个节点表示颜色值，end表示。。。（我觉得此处end表示延迟， end 为真表示此处有延迟，为假表示此处没有延迟，而当前节点的颜色表示当前区间各段的颜色）

此题比较好的是用位运算或模拟染色情况。

刚开始考虑用数组表示颜色，写着写着自己都不知道写的什么了。参考的大神的程序。

http://www.cnblogs.com/rainydays/archive/2011/05/14/2046419.html

注意：输入的区间两端点大小的问题。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;

#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn=100010;

int n, t, op;

struct node{
    int color;
    bool end;
}sum[maxn<<2];

void build(int l, int r, int rt){
    sum[rt].color = 1;
    sum[rt].end = true;
    if(l == r){
        return ;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
}

void Update(int L, int R, int c, int l, int r, int rt){
    if(L == l && r == R){
        sum[rt].end = true;
        sum[rt].color = c;
        return ;
    }
    if(sum[rt].end){
        sum[rt].end = false;
        sum[rt<<1].color = sum[rt].color;
        sum[rt<<1].end = true;
        sum[rt<<1|1].color = sum[rt].color;
        sum[rt<<1|1].end = true;
    }
    int m = (l+r)>>1;
    if(L <= m){
        if(R<=m) Update(L, R, c, lson);
        else Update(L, m, c, lson);
    }
    if(R > m){
        if(L<=m) Update(m+1, R, c, rson);
        else Update(L, R, c, rson);
    }
    sum[rt].color = sum[rt<<1].color | sum[rt<<1|1].color;
}

int Query(int L, int R, int l, int r, int rt){
    if(sum[rt].end){
        return sum[rt].color;
    }
    if(l == L && r == R){
        return sum[rt].color;
    }
    int m = (l+r)>>1;
    int ret = 0;
    if(L <= m){
        if(R <= m) Query(L, R, lson);
        else Query(L, m, lson);
    }
    if(R <= m)
        return Query(L, R, lson);
    else if(L > m)
        return Query(L, R, rson);
    else return Query(L, m, lson)|Query(m+1, R, rson);
}

int counter(int a){
    int x = 1;
    int ret = 0;
    for(int i=0; i<32; i++, x<<=1){
        if(x & a) ret++;
    }
    return ret;
}

int main(){
    freopen("in.txt", "r", stdin);
    char ch[2];
    int a, b, c;
    while(scanf("%d %d %d", &n, &t, &op) != EOF){
        build(1, n, 1);
        while(op--){
            scanf("%s", ch);
            if(ch[0] == 'C'){
                scanf("%d %d %d", &a, &b, &c);
                if(a > b) swap(a, b);
                Update(a, b, 1<<(c-1), 1, n, 1);
            }else if(ch[0] == 'P'){
                scanf("%d %d", &a, &b);
                if(a > b) swap(a, b);
                printf("%d\n", counter(Query(a, b, 1, n, 1)));
            }
        }
    }
    return 0;
}