HDU 3974 Assign the task(dfs時間戳+線段樹成段更新)
阿新 • • 發佈:2019-01-05
題意:給定點的上下級關係,規定如果給i分配任務a,那麼他的所有下屬。都停下手上的工作,開始做a。
操作 T x y 分配x任務y,C x詢問x的當前任務;
Sample Input
1 5 4 3 3 2 1 3 5 2 5 C 3 T 2 1 C 3 T 3 2 C 3Sample Output
Case #1: -1 1 2思路:
利用dfs深度優先遍歷重新編號,使一個結點的兒子連續。然後成段更新。
程式碼:
#include<iostream> #include<cstdio> #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 #define read_ freopen("i.txt","r",stdin) using namespace std; const int N=50010; int task[N<<2],lazy[N<<2]; struct node{ int to,next; }edge[N]; int ll; int l[N],r[N]; int uset[N],adj[N]; int num; void addedge(int u,int v) { edge[ll].to=u; edge[ll].next=adj[v]; adj[v]=ll++; } void dfs(int root) { int i=adj[root]; l[root]=(++num); while(i!=-1) { dfs(edge[i].to); i=edge[i].next; } r[root]=num; } void pushdown(int rt) { if(lazy[rt]!=-1) { task[rt<<1]=lazy[rt]; task[rt<<1|1]=lazy[rt]; lazy[rt<<1]=lazy[rt]; lazy[rt<<1|1]=lazy[rt]; lazy[rt]=-1; } } int find_set(int x) { if(uset[x]!=x) uset[x]=find_set(uset[x]); return uset[x]; } void build_tree(int l,int r,int rt) { task[rt]=-1; lazy[rt]=-1; if(l==r) return; int m=(l+r)>>1; build_tree(lson); build_tree(rson); } void update(int L,int R,int val,int l,int r,int rt) { if(L<=l&&r<=R) { task[rt]=val; lazy[rt]=val; return; } pushdown(rt); int m=(l+r)>>1; if(L<=m) update(L,R,val,lson); if(R>m) update(L,R,val,rson); } void query(int L,int R,int l,int r,int rt) { if(L<=l&&r<=R) { printf("%d\n",task[rt]); return ; } pushdown(rt); int m=(l+r)>>1; if(L<=m) query(L,R,lson); if(R>m) query(L,R,rson); } int main() { //read_; int T; scanf("%d",&T); for(int cas=1;cas<=T;cas++) { int n; scanf("%d",&n); ll=0; num=0; for(int i=0;i<=n;i++) { adj[i]=-1; uset[i]=i; } int u,v; for(int i=1;i<n;i++) { scanf("%d%d",&u,&v); uset[u]=v; addedge(u,v); } int root=find_set(1); dfs(root); build_tree(1,num,1); int m; scanf("%d",&m); char op[5]; int a,b; printf("Case #%d:\n",cas); for(int i=0;i<m;i++) { scanf("%s",op); if(op[0]=='C') { scanf("%d",&a); query(l[a],l[a],1,num,1); } else{ scanf("%d%d",&a,&b); update(l[a],r[a],b,1,num,1); } } } return 0; }