HDU 3974 Assign the task（dfs时间戳+线段树成段更新）

题意：给定点的上下级关系，规定假设给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 3

Sample 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
;

int ll;
int l
,r
;
int uset
,adj
;
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;
}