HDU 3974(Assign the task)线段树维护dfs序

题目:
在这里啊
题意:
一个公司有一个老板及许多员工 ,除老板外,每个员工都有自己的领导,然而每次分配工作之后,如果自己的领导得到了工作则自己也会去干相同的工作。
每次询问要问一个一人在干什么工作。
思路:
最简单可以想到的就是单点维护单点查询,这个我帮你们试过了,妥妥超时,
所以我们要优化一下,要区间更新,单点查询,因为每个人都有自己的领导,所以从公司的老板开始做深搜,这样子之后统计一下每个点的开始进入搜索的点编号和从这个点出去的点编号,就可以得到这点如果被更新时,相应的应该更新的所有点的编号都在一个区间内,只需要更新这个区间即可。

查询就是简单的单点查询,细节看代码注释吧。

/*用线段树维护关系树的dfs序,区间更新单点查询*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=5e4+10;
struct node{
    int b,e,v;
}t[maxn*4];
struct edge{
    int u,v,next;
}e[maxn];
int head[maxn];     //广搜要建图,使用链式前向星
int per[maxn];     //记录每个点领导,主要是为了找出这个公司老板  开始广搜
int num[maxn];      //记录dfs顺序
int cont,esum;      
int sta[maxn],en[maxn];    //记录一个点开始的 搜索点  以及最后一个搜索点
int vis[maxn];       //广搜标记数组,看是否被搜过。
void init(){          //初始化
    for(int i=0;i<maxn;i++){
        per[i]=i;
        vis[i]=0;
        head[i]=-1;
    }
    cont=esum=0;
}
void dfs(int x){         //dfs找到正确的dfs序
    num[x]=++cont;
    vis[x]=1;
    sta[x]=cont;
    for(int i=head[x];i!=-1;i=e[i].next){
        if(!vis[e[i].v]){
            dfs(e[i].v);
        }
    }
    en[x]=cont;
    return ;
}
void add_edge(int uu, int vv){      //链式前向星加边
    e[++esum].u=uu;e[esum].v=vv;
    e[esum].next=head[uu];
    head[uu]=esum;
} 
void push_down(int node){           //向下维护线段树
    if(t[node].v!=-1){
        t[node<<1].v=t[node<<1|1].v=t[node].v;
        t[node].v=-1;
    }
}
void create(int bb,int ee,int node){      //建树
    t[node].b=bb;
    t[node].e=ee;
    t[node].v=-1;
    if(bb==ee)return ;
    int mid=(bb+ee)>>1;
    create(bb,mid,node<<1);
    create(mid+1,ee,node<<1|1);
}
void update(int bb,int ee,int node,int val){       //更新
    if(bb<=t[node].b&&ee>=t[node].e){
        t[node].v=val;
        return ;
    }
    int mid=(t[node].b+t[node].e)>>1;
    push_down(node);
    if(bb>mid)update(bb,ee,node<<1|1,val);
    else if(ee<=mid)update(bb,ee,node<<1,val);
    else{
        update(bb,mid,node<<1,val);
        update(mid+1,ee,node<<1|1,val);
    }
}
int query(int pos,int node){       //查找
    if(t[node].b==t[node].e){
        return t[node].v;
    }
    int mid=(t[node].b+t[node].e)>>1;
    push_down(node);
    if(pos<=mid)return query(pos,node<<1);
    else return query(pos,node<<1|1);
}
int main()
{
    int t,n,m,cnt=0;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        create(1,n,1);
        init();
        for(int i=0;i<n-1;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            per[u]=v;
            add_edge(v,u);
        }
        for(int i=1;i<=n;i++)
        if(per[i]==i){
            dfs(i);
            break;
        }
        scanf("%d",&m);
        printf("Case #%d:\n",++cnt);
        for(int i=0;i<m;i++){
            char ch[2];
            scanf("%s",ch);
            if(ch[0]=='C'){
                int x;
                scanf("%d",&x);
                printf("%d\n",query(num[x],1));
            }
            if(ch[0]=='T'){
                int x,y;
                scanf("%d%d",&x,&y);
                update(sta[x],en[x],1,y);
            }
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值