HDU 3974 线段树（将树映射到区间）

#include "stdio.h" #include "string.h" struct node { int l,r,val,lazy; }data[200010]; struct Edge { int to,next; }edge[50010]; int tot,cnt; int head[50010],start[50010],end[50010],used[50010]; void addedge(int x,int y) { edge[tot].to=x; edge[tot].next=head[y]; head[y]=tot++; } void dfs(int x) { int i; cnt++; start[x]=cnt; // BOSS的起始编号 for (i=head[x];i!=⑴;i=edge[i].next) dfs(edge[i].to); end[x]=cnt; // 该BOSS的最后1名员工的编号 } void build(int l,int r,int k) { int mid; data[k].l=l; data[k].r=r; data[k].val=⑴; data[k].lazy=0; if (l==r) return ; mid=(l+r)/2; build(l,mid,k*2); build(mid+1,r,k*2+1); } void Pushdown(int k) { if (data[k].lazy==0) return ; if (data[k].l==data[k].r) return; data[k*2].lazy=data[k*2+1].lazy=data[k].lazy; data[k*2].val=data[k*2+1].val=data[k].lazy; data[k].lazy=0; } void updata(int l,int r,int k,int op) { int mid; if (data[k].l==l && data[k].r==r) { data[k].val=op; data[k].lazy=op; return ; } Pushdown(k); mid=(data[k].l+data[k].r)/2; if (r<=mid) updata(l,r,k*2,op); else if (l>mid) updata(l,r,k*2+1,op); else { updata(l,mid,k*2,op); updata(mid+1,r,k*2+1,op); } if (data[k*2].val==data[k*2+1].val) data[k].val=data[k*2].val; else data[k].val=⑴; } int query(int x,int k) { int mid; if (data[k].val!=⑴) return data[k].val; if (data[k].l==data[k].r) return data[k].val; Pushdown(k); mid=(data[k].l+data[k].r)/2; if (x<=mid) return query(x,k*2); else return query(x,k*2+1); } int main() { int T,Case,i,x,y,m,n; char ch[10]; scanf("%d",&T); Case=0; while (T--) { Case++; printf("Case #%d: ",Case); memset(used,0,sizeof(used)); // 查找根节点 tot=0;// 表示当前边 cnt=0;// 表示当前点 memset(head,⑴,sizeof(head)); scanf("%d",&n); for (i=1;i<n;i++) { scanf("%d%d",&x,&y); used[x]=1; addedge(x,y); // 建边 } for (i=1;i<=n;i++) if (used[i]==0) { dfs(i); // 重新对各点进行编号，以便将原本的树映照到区间 break; } build(1,cnt,1); scanf("%d",&m); while (m--) { scanf("%s",ch); if (ch[0]=='C') { scanf("%d",&x); printf("%d ",query(start[x],1)); } else { scanf("%d%d",&x,&y); updata(start[x],end[x],1,y); } } } return 0; }