題意

給我們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 线段树染色问题+延时标记的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。