hdu3974-Assign the task-（dfs+线段树）

Posted 2022-04-30 shoulinniao

题意：有n个人，有上下级关系，有m个操作，有两种操作1.把一个任务分给某个人，他的下属也会停下手中工作和他一起做；2.查询某个人的当前任务是什么？

解题：n-1个关系，总有一个人没有上级，以他为根节点用dfs搜索整张图可以得到一棵树，按“根左右”先序遍历，根表示自己，遍历到最右边的儿子结束，这段区间为自己的管辖范围，按遍历顺序构造一棵线段树，每个人记录管辖范围，一有任务分给某人，他就分散下去，线段树区间更新。

  1 #include<stdio.h>
  2 #include<iostream>
  3 #include<algorithm>
  4 #include<cstring>
  5 #include<math.h>
  6 #include<string>
  7 #include<map>
  8 #include<queue>
  9 #include<stack>
 10 #include<set>
 11 #define ll long long
 12 #define inf 0x3f3f3f3f
 13 using namespace std;
 14 
 15 int T,n,v,u,cnt,m,x,y;
 16 char s[10];
 17 vector<int>son[50005];///儿子可能有多个,邻接表
 18 int dad[50005];///下标是儿子，内容是父亲
 19 int l[50005];
 20 int r[50005];///l和r两个数组之间的范围 表示 管辖的员工在线段树中的编号范围
 21 int tree[50005*4];
 22 int lazy[50005*4];
 23 
 24 void dfs(int now)///now是当前搜到的员工的编号
 25 
 26     l[now]=++cnt;
 27     for(int i=0;i<son[now].size();i++)
 28     
 29         int xx=son[now][i];
 30         dfs(xx);
 31     
 32     r[now]=cnt;
 33 
 34 
 35 void pushdown(int rt)
 36 
 37     if( lazy[rt]!=-1 )
 38     
 39         tree[ rt*2 ] = tree[ rt*2+1 ] = lazy[ rt*2 ] = lazy[ rt*2+1 ] =lazy[rt];
 40         lazy[rt]=-1;///懒标记赋值给儿子后清空为-1
 41     
 42 
 43 
 44 
 45 void build(int l,int r,int rt)
 46 
 47     if(l==r)
 48     
 49         tree[rt]=-1;
 50         return;
 51     
 52     int mid=(l+r)/2;
 53     build(l,mid,rt*2);
 54     build(mid+1,r,rt*2+1);///不需要求和
 55 
 56 
 57 void update(int L,int R,int p,int l,int r,int rt)
 58 
 59     if( L<=l && r<=R )
 60     
 61         tree[rt]=lazy[rt]=p;
 62         return;
 63     
 64     int mid=(l+r)/2;
 65     pushdown(rt);
 66     if( L<=mid )
 67         update(L,R,p,l,mid,rt*2);
 68     if( mid+1<=R )
 69         update(L,R,p,mid+1,r,rt*2+1);
 70 
 71 
 72 int query(int l,int r,int rt,int x)
 73 
 74     if( l==r )
 75         return tree[rt];
 76     int mid=(l+r)/2;
 77     pushdown(rt);
 78     if(x<=mid)
 79         return query(l,mid,rt*2,x);
 80     else
 81         return query(mid+1,r,rt*2+1,x);
 82 
 83 
 84 int main()
 85 
 86 
 87     scanf("%d",&T);
 88     for(int t=1;t<=T;t++)
 89     
 90         memset(dad,0,sizeof(dad));
 91         memset(lazy,-1,sizeof(lazy));
 92         memset(tree,-1,sizeof(tree));
 93         for(int i=1;i<=n;i++)
 94             son[i].clear();
 95 
 96         scanf("%d",&n);
 97         for(int i=1;i<n;i++)
 98         
 99             scanf("%d%d",&u,&v);///v是u的父亲
100             dad[u]=v;
101             son[v].push_back(u);
102         
103 
104         for(int i=1;i<=n;i++)
105         
106             if( dad[i]==0 )///没有父亲，便是祖宗
107             
108                 v=i;
109                 break;
110             
111         
112         cnt=0;
113         dfs(v);///从祖宗开始往下搜
114 
115         printf("Case #%d:\n",t);
116         scanf("%d",&m);
117         while(m--)
118         
119             string ss;
120             cin>>ss;
121             if( ss[0]==‘C‘ )///查询a的任务
122             
123                 scanf("%d",&x);
124                 int ans=query(1,cnt,1,l[x]);
125                     printf("%d\n",ans);
126             
127             else
128             
129                 scanf("%d%d",&x,&y);///让员工a做任务b，a会叫他的下属一起做，左右区间表示需要做的人，包括自己
130                 update( l[x],r[x],y,1,cnt,1 );
131             
132         
133 
134     
135     return 0;
136