国内最全IT社区平台 联系我们 | 收藏本站
华晨云阿里云优惠2
您当前位置:首页 > php开源 > php教程 > HDU 3974 线段树(将树映射到区间)

HDU 3974 线段树(将树映射到区间)

来源:程序员人生   发布时间:2015-03-28 08:16:47 阅读次数:2518次

第1次写将树映照到区间的线段树。。。

线段树部份很简单

主要是将原本的关系树根据BOSS关系重新编号

以便把每一个BOSS所带领的员工全部压入1个连续区间内

然后记录每一个BOSS的起始编号和他的最后1名的员工的编号

然后用线段树成端更新,单点查找便可


#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; }


生活不易,码农辛苦
如果您觉得本网站对您的学习有所帮助,可以手机扫描二维码进行捐赠
程序员人生
------分隔线----------------------------
分享到:
------分隔线----------------------------
关闭
程序员人生