1. 程式人生 > >HDU 3974 Assign the task(dfs時間戳+線段樹成段更新)

HDU 3974 Assign the task(dfs時間戳+線段樹成段更新)

題意:給定點的上下級關係,規定如果給i分配任務a,那麼他的所有下屬。都停下手上的工作,開始做a。

          操作 T x y 分配x任務y,C x詢問x的當前任務;

Sample Input

1 5 4 3 3 2 1 3 5 2 5 C 3 T 2 1 C 3 T 3 2 C 3

Sample Output

Case #1: -1 1 2

思路:

利用dfs深度優先遍歷重新編號,使一個結點的兒子連續。然後成段更新。

程式碼:

#include<iostream>
#include<cstdio>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define read_ freopen("i.txt","r",stdin)
using  namespace std;
const int N=50010;

int task[N<<2],lazy[N<<2];
struct node{
        int to,next;
}edge[N];

int ll;
int l[N],r[N];
int uset[N],adj[N];
int num;
void addedge(int u,int v)
{
        edge[ll].to=u;
        edge[ll].next=adj[v];
        adj[v]=ll++;
}


void dfs(int root)
{
        int i=adj[root];
        l[root]=(++num);
        while(i!=-1)
        {
                dfs(edge[i].to);
                i=edge[i].next;
        }
        r[root]=num;
}

void pushdown(int rt)
{
        if(lazy[rt]!=-1)
        {
                task[rt<<1]=lazy[rt];
                task[rt<<1|1]=lazy[rt];
                lazy[rt<<1]=lazy[rt];
                lazy[rt<<1|1]=lazy[rt];
                lazy[rt]=-1;
        }

}
int find_set(int x)
{
        if(uset[x]!=x)
                uset[x]=find_set(uset[x]);
        return uset[x];
}

void build_tree(int l,int r,int rt)
{
        task[rt]=-1;
        lazy[rt]=-1;
        if(l==r)
                return;
        int m=(l+r)>>1;
        build_tree(lson);
        build_tree(rson);
}

void update(int L,int R,int val,int l,int r,int rt)
{
        if(L<=l&&r<=R)
        {
                task[rt]=val;
                lazy[rt]=val;
                return;
        }
        pushdown(rt);
        int m=(l+r)>>1;
        if(L<=m) update(L,R,val,lson);
        if(R>m) update(L,R,val,rson);
}

void query(int L,int R,int l,int r,int rt)
{
        if(L<=l&&r<=R)
        {
                printf("%d\n",task[rt]);
                return ;
        }
        pushdown(rt);
        int m=(l+r)>>1;
        if(L<=m) query(L,R,lson);
        if(R>m) query(L,R,rson);
}

int main()
{
        //read_;
        int T;
        scanf("%d",&T);
        for(int cas=1;cas<=T;cas++)
        {
                int n;
                scanf("%d",&n);
                ll=0;
                num=0;
                for(int i=0;i<=n;i++)
                {
                        adj[i]=-1;
                        uset[i]=i;
                }
                int u,v;
                for(int i=1;i<n;i++)
                {
                        scanf("%d%d",&u,&v);
                        uset[u]=v;
                        addedge(u,v);
                }
                int root=find_set(1);
                dfs(root);
                build_tree(1,num,1);
                int m;
                scanf("%d",&m);
                char op[5];
                int a,b;
                printf("Case #%d:\n",cas);
                for(int i=0;i<m;i++)
                {
                        scanf("%s",op);
                        if(op[0]=='C')
                        {
                                scanf("%d",&a);
                               
                                query(l[a],l[a],1,num,1);
                        }
                        else{
                                scanf("%d%d",&a,&b);
                                update(l[a],r[a],b,1,num,1);
                        }
                }
        }
        return 0;
}