Assign the task HDU - 3974(dfs序+线段树)

思路

将员工之间的关系建图,然后对图dfs将顺序记录下来(一个节点进去一次出来一次),再进行区间维护即可.

代码

#include<map>
#include<set>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
#include<iomanip>
#include<sstream>
#include<iostream>
#include<algorithm>
#define INF 0x3f3f3f3f
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define ll __int64
//#define int ll
//typedef   long long ll;
typedef unsigned long long ull;
const int MAXN=5e4+5;
const int MOD=2333;
const double eps=1e-6;
using namespace std;
//-------------------------------------------//
int lazy[MAXN<<2],s[MAXN],e[MAXN],vis[MAXN],n,m;
vector<int> g[MAXN];
void pushdown(int rt)
{
    if(lazy[rt]!=-1)
    {
        lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
        lazy[rt]=-1;
    }
}
void updata(int x,int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        lazy[rt]=x;
        return;
    }
    pushdown(rt);
    int m=l+r>>1;
    if(L<=m)updata(x,L,R,ls);
    if(m<R)updata(x,L,R,rs);
}
int query(int pos,int l,int r,int rt)
{
    if(l==r)return lazy[rt];
    pushdown(rt);
    int m=l+r>>1;
    if(m>=pos)return query(pos,ls);
    else return query(pos,rs);
}
int cnt;
void dfs(int u)
{
    s[u]=++cnt;
    for(int i=0;i<g[u].size();++i)dfs(g[u][i]);
    e[u]=++cnt;
}
int main()
{
    int t;
    cin>>t;
    for(int cas=1;cas<=t;++cas)
    {
        cnt=0;
        printf("Case #%d:\n",cas);
        memset(lazy,-1,sizeof lazy);
        memset(vis,0,sizeof vis);
        scanf("%d",&n);
        for(int i=0;i<=n;++i)g[i].clear();
        for(int i=0;i<n-1;++i)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[v].push_back(u);
            vis[u]++;
        }
        for(int i=1;i<=n;++i)
        if(!vis[i])
        {
            dfs(i);
            break;
        }
        scanf("%d",&m);
        while(m--)
        {
            char op;
            int x,y;
            scanf(" %c",&op);
            if(op=='T')
            {
                scanf("%d%d",&x,&y);
                updata(y,s[x],e[x],1,cnt,1);
            }
            else
            {
                scanf("%d",&x);
                printf("%d\n",query(s[x],1,cnt,1));
            }
        }
    }
    return 0;
}