HDU-5023 线段树染色问题+延时标记

题意

给我们1-n的区间段 两种操作
P a b c　表示把a到b染成c颜色
Q a b 表示求a到b的颜色种类数量
注意初始全部先染成颜色2

分析

线段树经典染色问题
可以用二进制来表示颜色数量 一共30个颜色 那么最多也就是30位 不会超过int的范围 二进制下第x位为1表示有x+1的颜色
父节点可以用”|”运算来统计两个子节点的信息

CODE

#include<bits/stdc++.h> #define lson l,mid,rt<<1 #define rson mid+1,r,rt<<1|1 using namespace std; typedef long long ll; const int maxn = 1000010; struct tree{int color,tag;//延时标记 }T[maxn<<2]; void build(int l,int r,int rt) {T[rt].tag=0;//注意这里要初始化树中的每个点都要初始化成0 才不会出错if(l==r){T[rt].color=2;return;}int mid = (l+r)>>1;build(lson);build(rson);T[rt].color = T[rt<<1].color | T[rt<<1|1].color; } void push_down(int l,int r,int rt){if(T[rt].tag){T[rt<<1].color = T[rt<<1|1].color = T[rt].tag;T[rt<<1].tag = T[rt<<1|1].tag = T[rt].tag;T[rt].tag=0;} } void update(int s,int e,int cor,int l,int r,int rt){if(s<=l&&e>=r){T[rt].color = 1<<(cor-1);T[rt].tag = 1<<(cor-1);return;}push_down(l,r,rt);int mid = (l+r)>>1;if(s<=mid)update(s,e,cor,lson);if(e>=mid+1)update(s,e,cor,rson);T[rt].color = T[rt<<1].color|T[rt<<1|1].color; } int query(int s,int e,int l,int r,int rt){if(s<=l&&e>=r)return T[rt].color;push_down(l,r,rt);int mid = l+r>>1;int ans=0;if(s<=mid)ans|=query(s,e,lson);if(e>=mid+1)ans|=query(s,e,rson);return ans; } int main() {int n,m;while(scanf("%d%d",&n,&m)){if(n==0&&m==0)return 0;char cc[3];build(1,n,1);while(m--){int a,b,c;scanf(" %s",cc);if(cc[0]=='P'){scanf("%d%d%d",&a,&b,&c);update(a,b,c,1,n,1);}else{scanf("%d%d",&a,&b);int ans = query(a,b,1,n,1);bool f=0;while(ans){if(ans&1){if(f)printf(" %d",cnt);else printf("%d",cnt),f=1;}cnt++;ans>>=1;}puts("");}}}return 0; }

总结

以上是生活随笔为你收集整理的HDU-5023 线段树染色问题+延时标记的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。