题意
一个公司有n个人,有一个人是领导(等级最高)没有上司,其余每个人都有一个上司。
给定m个要求,当输入为“C x”时,输出x当前的任务;当输入为"T x y"时,给x分配一个任务y(此时,x和x的下属都需要放弃原来的任务,执行当前的任务y)。
思路
dfs序+区间更新
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=131072;
int in[maxn/2],out[maxn/2],sum[maxn];
int head[maxn],pre[maxn/2];
int time,cnt,root;
struct node
{
int to,next;
}e[maxn];
void add(int u,int v)
{
e[cnt]=node{v,head[u]};
head[u]=cnt++;
}
void dfs(int x,int fa)
{
in[x]=++time;
for(int i=head[x];i!=-1;i=e[i].next)
{
int a=e[i].to;
if(e[i].to==fa)
continue;
dfs(a,x);
}
out[x]=time;
}
void pushup(int rt)
{
if(sum[rt]!=-1) //有任务时才更新
{
sum[rt<<1]=sum[rt<<1|1]=sum[rt];
sum[rt]=-1;
}
}
void build(int l,int r,int rt)
{
sum[rt]=-1;
if(l==r)
return ;
int m=(l+r)>>1;
build(l,m,rt<<1);
build(m+1,r,rt<<1|1);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l && R>=r)
{
sum[rt]=c; //分配任务
return ;
}
pushup(rt);
int m=(l+r)>>1;
if(L<=m)
update(L,R,c,l,m,rt<<1);
if(R>m)
update(L,R,c,m+1,r,rt<<1|1);
}
int query(int L,int l,int r,int rt)
{
if(l==r)
return sum[rt];
pushup(rt);
int m=(l+r)>>1;
int ans;
if(L<=m)
ans=query(L,l,m,rt<<1);
else
ans=query(L,m+1,r,rt<<1|1);
return ans;
}
int main()
{
int t,n,m;
scanf("%d",&t);
for(int k=1;k<=t;k++)
{
time=cnt=0;
memset(head,-1,sizeof(head));
int x,y;
scanf("%d",&n);
for(int i=1;i<=n;i++)
pre[i]=i;
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
pre[x]=y;
add(y,x);
if(pre[y]==y) //找出领导(根节点)
root=y;
}
dfs(root,-1);
build(1,n,1);
printf("Case #%d:\n",k);
scanf("%d",&m);
char s;
for(int i=1;i<=m;i++)
{
scanf(" %c",&s);
if(s=='C')
{
scanf("%d",&x);
printf("%d\n",query(in[x],1,n,1));
}
else
{
scanf("%d%d",&x,&y);
update(in[x],out[x],y,1,n,1);
}
}
}
return 0;
}
一开始理解的并不好,写的程序比较乱,虽然提交过了,但再次查看时发现有许多没有用的代码。还是需要好好理解线段树。。。