HDU 3974 線段樹(將樹映射到區間)
來源:程序員人生 發布時間:2015-03-28 08:16:47 閱讀次數:2709次
第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;
}
生活不易,碼農辛苦
如果您覺得本網站對您的學習有所幫助,可以手機掃描二維碼進行捐贈