Assign the task HDU - 3974（線段樹維護DFS序）

鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=3974

題意：給定一棵樹，維護兩種操作：C x 表示查詢 x 當(dāng)前的顏色。T x y 表示將包括 x 在內(nèi)的子樹染成顏色 y 。

思路：線段樹維護DFS序

#include <cstdio> #include <algorithm> #include <cstring> #define ls (rt<<1) #define rs (rt<<1|1) #define ll long long using namespace std; const int maxn=5e4+5;int t,n; vector<int> e[maxn]; int in[maxn]; int dfn[maxn],sz[maxn],times=0;void dfs(int u) {dfn[u]=++times;sz[u]=1;for(auto v: e[u]){dfs(v);sz[u]+=sz[v];} }int st[maxn<<2]; void build(int rt,int L,int R) {st[rt]=-1;if(L==R) return;int mid=(L+R)>>1;build(ls,L,mid);build(rs,mid+1,R); }void pushDown(int rt) {if(st[rt]!=-1){st[ls]=st[rs]=st[rt];st[rt]=-1;} }void update(int rt,int l,int r,int L,int R,int v) {if(l<=L&&R<=r){st[rt]=v;return;}pushDown(rt);int mid=(L+R)>>1;if(l<=mid) update(ls,l,r,L,mid,v);if(r>mid) update(rs,l,r,mid+1,R,v); }int query(int rt,int p,int L,int R) {if(L==R) return st[rt];pushDown(rt);int mid=(L+R)>>1;if(p<=mid) return query(ls,p,L,mid);if(p>mid) return query(rs,p,mid+1,R); }int main() {int Case=0;scanf("%d",&t);while(t--){scanf("%d",&n);for(int i=1; i<=n; ++i)dfn[i]=sz[i]=in[i]=0,e[i].clear();times=0;for(int i=1; i<=n-1; ++i){int u,v;scanf("%d%d",&u,&v);e[v].push_back(u);in[u]++;}for(int i=1; i<=n; ++i)if(in[i]==0) dfs(i);build(1,1,n);int q;scanf("%d",&q);printf("Case #%d:\n",++Case);while(q--){char op[5];int x,y;scanf("%s",op);if(op[0]=='C'){scanf("%d",&x);printf("%d\n",query(1,dfn[x],1,n));}else{scanf("%d%d",&x,&y);update(1,dfn[x],dfn[x]+sz[x]-1,1,n,y);}}}return 0; }

總結(jié)

以上是生活随笔為你收集整理的线段树习题的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。