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);
    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=-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 线段树（将树映射到区间）

标签：

原文地址：http://blog.csdn.net/u011932355/article/details/44065089