题目来源:http://acm.hdu.edu.cn/showproblem.php?pid=3974
★开学之后又有好久没写博客了,快一个月了,写了一下以前写过的题目,居然忘记怎么写了。。。
不会写的原因是MLE 没有把head数组清0,调了一晚上,枯了
题意:
现在有一个公司,里面有n个员工,员工之间有上级和下属的关系。上级得到了任务 他会立刻和其所有下属转手干这份工作(把之前的事抛在一边)。然后给你两个操作,①:给某个员工一个工作 ②:问某个员工在干什么工作
思路:
在dfs遍历树的过程中 记录下每个父节点的起始时间和终止时间
(点左边的数字是起始时间,右边的数字是终止时间)
然后很容易发现,起始时间 和 终止时间 之间的点就是 父节点的子节点
所以我们更新的时候在这个区间更新就好了
单点查询的时候,查询 起始点 终止点 都行
剩下的就是基本的线段树操作了~详见代码
代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<map>
#include<stack>
#include<queue>
#include<set>
#include<vector>
#define ls k<<1,l,mid
#define rs k<<1|1,mid+1,r
using namespace std;
typedef long long LL;
const int maxn=5e4+5;
const int mod=1e9+7;
const int inf=0x7ffff;
const double pi=acos(-1);
const double eps=1e-8;
int n,m,cnt;
struct node
{
int to,nex;
}g[maxn];
int head[maxn],st[maxn],en[maxn];
int v[maxn<<2],lazy[maxn<<2];
bool vis[maxn];
template<class t>
inline void read(t &x)
{
char c;x=1;
while((c=getchar())<'0'||c>'9') if(c=='-') x=-1;
t res=c-'0';
while((c=getchar())>='0'&&c<='9') res=res*10+c-'0';
x*=res;
}
void addedge(int x,int y)
{
g[++cnt].to=y;
g[cnt].nex=head[x];
head[x]=cnt;
}
void dfs(int k)
{
// cout<<k<<'x'<<endl;
st[k]=++cnt;
for(int i=head[k];i;i=g[i].nex){
dfs(g[i].to);
}
en[k]=cnt;
}
void build(int k,int l,int r)
{
lazy[k]=0;
v[k]=-1;
if(l==r) return ;
int mid=l+r>>1;
build(ls); build(rs);
}
void push_down(int k,int l,int r,int mid)
{
if(lazy[k]){
lazy[k<<1]=lazy[k<<1|1]=lazy[k];
v[k<<1]=v[k<<1|1]=lazy[k];
lazy[k]=0;
}
}
void update(int k,int l,int r,int x,int y,int val)
{
if(x<=l&&r<=y){
v[k]=val;
lazy[k]=val;
return ;
}
int mid=l+r>>1;
push_down(k,l,r,mid);
if(mid>=x) update(ls,x,y,val);
if(mid<y) update(rs,x,y,val);
}
int query(int k,int l,int r,int pos)
{
if(l==r) return v[k];
int mid=l+r>>1;
push_down(k,l,r,mid);
if(mid>=pos) return query(ls,pos);
return query(rs,pos);
}
int main()
{
int t;
read(t);
int cas=0;
while(t--){
read(n);
printf("Case #%d:\n",++cas);
memset(vis,0,sizeof vis);
memset(head,0,sizeof head);
cnt=0;
for(int i=1;i<n;i++){
int a,b;
read(a); read(b);
addedge(b,a);
vis[a]=1;
}
cnt=0;
for(int i=1;i<=n;i++){
if(!vis[i]){
dfs(i);
break;
}
}
build(1,1,cnt);
char s[10];
int p,q;
read(m);
// for(int i=1;i<=n;i++){
// cout<<st[i]<<' '<<en[i]<<endl;
// }
while(m--){
scanf("%s %d",s,&p);
if(s[0]=='C'){
printf("%d\n",query(1,1,cnt,st[p]));
}
else{
read(q);
update(1,1,cnt,st[p],en[p],q);
}
}
}
return 0;
}