本文主要是介绍HDU 3974 线段树(将树映射到区间),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
第一次写将树映射到区间的线段树。。。
线段树部分很简单
主要是将原有的关系树根据BOSS关系从新编号
以便把每个BOSS所带领的员工全部压入一个连续区间内
然后记录每个BOSS的起始编号和他的最后一名的员工的编号
然后用线段树成端更新,单点查找即可
#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!=-1;i=edge[i].next)dfs(edge[i].to);end[x]=cnt; // 该BOSS的最后一名员工的编号
}void build(int l,int r,int k)
{int mid;data[k].l=l;data[k].r=r;data[k].val=-1;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);elseif (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=-1;
}int query(int x,int k)
{int mid;if (data[k].val!=-1) 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:\n",Case);memset(used,0,sizeof(used)); // 查找根节点tot=0;// 表示当前边cnt=0;// 表示当前点memset(head,-1,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\n",query(start[x],1));}else{scanf("%d%d",&x,&y);updata(start[x],end[x],1,y);}}}return 0;
}
这篇关于HDU 3974 线段树(将树映射到区间)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!