HDOJ4366解题报告【dfs序+线段树】

题目地址：

　　http://acm.hdu.edu.cn/showproblem.php?pid=4366

题目概述：

　　给出一个公司所有员工的上下级关系（数据保证是一棵树），现在想解雇一些员工，而他们的位置由他们的手下里能力值比他大而且忠诚度最高的人来替代，给出m个询问，求输出替代他们的人的编号。

大致思路：

　　学习了一下dfs序的用法，附学习的博客：http://blog.csdn.net/Miracle_ma/article/details/51485118

　　简单的说dfs序就是将一棵树变成线性结构的算法，并且保证点i的子树都在i的后面。

　　根据dfs序首先将员工之间的关系转化成数组，然后对于员工的数据a按照能力值降序排列，并且将所有的询问q也按对应的能力值降序排列。

　　排完序之后依次插入a中的元素进入线段树中，比较当前插入的a_i与未处理的q中能力值最高的比较，如果a_i的能力值比较小，则处理这个询问，否则继续插入。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <ctime>
#include <map>
#include <stack>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;

#define sacnf scanf
#define scnaf scanf
#define maxn  50010
#define maxm 26
#define inf 1061109567
#define Eps 0.00001
const double PI=acos(-1.0);
#define mod 7
#define MAXNUM 10000
void Swap(int &a,int &b) {int t=a;a=b;b=t;}
int Abs(int x) {return (x<0)?-x:x;}
typedef long long ll;
typedef unsigned int uint;

struct node
{
    int pos,abi,loy;
    bool operator < (const node &a) const
    {
        if(abi==a.abi) return loy>a.loy;
        return abi>a.abi;
    }
} a[maxn];

struct node1
{
    int pos,val;
} q[maxn];

vector<int> G[maxn];
int tme;
int st[maxn],ed[maxn];
int tabi[maxn],tloy[maxn];
int ans[maxn];
int tree[maxn*4];          ///loy最大的点的编号

bool cmp(node1 a,node1 b)
{
    return tabi[a.val]>tabi[b.val];
}

void dfs(int u,int fa)
{
    st[u]=++tme;
    int len=G[u].size();
    for(int i=0;i<len;i++)
    {
        int v=G[u][i];
        if(v!=fa) dfs(v,u);
    }
    ed[u]=tme;
}

void build_tree(int l,int r,int dir)
{
    if(l==r)
    {
        tree[dir]=1;
        return;
    }
    int m=(l+r)>>1;
    build_tree(l,m,dir<<1);
    build_tree(m+1,r,dir<<1|1);
    tree[dir]=1;
}

void add(int l,int r,int dir,int pos,int val)
{
    if(l==r)
    {
        tree[dir]=(tloy[tree[dir]]<tloy[val])?val:tree[dir];
        return;
    }
    int m=(l+r)>>1;
    if(pos<=m) add(l,m,dir*2,pos,val);
    else add(m+1,r,dir*2+1,pos,val);
    tree[dir]=(tloy[tree[dir*2]]<tloy[tree[dir*2+1]])?tree[dir*2+1]:tree[dir*2];
}

int query(int l,int r,int dir,int ql,int qr)
{
    if(ql>qr) return 1;
    if(ql>r||l>qr) return 1;
    if(ql<=l&&r<=qr) return tree[dir];
    int m=(l+r)>>1,t1,t2;
    t1=query(l,m,dir<<1,ql,qr);
    t2=query(m+1,r,dir<<1|1,ql,qr);
    return (tloy[t1]<tloy[t2])?t2:t1;
}

int main()
{
    //freopen("data.in","r",stdin);
    //freopen("data.out","w",stdout);
    //clock_t st=clock();
    int n,m,T;scanf("%d",&T);tree[0]=-1;
    while(T--)
    {
        sacnf("%d%d",&n,&m);
        int x;

        a[1].pos=1;a[1].abi=a[1].loy=-1;
        tloy[1]=tabi[1]=-1;
        for(int i=1;i<=n;i++) G[i].clear();

        for(int i=2;i<=n;i++)
        {
            scanf("%d%d%d",&x,&a[i].loy,&a[i].abi);
            x++;a[i].pos=i;tabi[i]=a[i].abi;tloy[i]=a[i].loy;
            G[x].push_back(i);G[i].push_back(x);
        }

        tme=0;
        dfs(1,-1);
        build_tree(1,n,1);

        sort(a+1,a+1+n);       ///按ability降序排列    a-->第i个人的数据

        for(int i=1;i<=m;i++)                  ///q-->第i个询问的序号及询问编号
        {
            scanf("%d",&q[i].val);q[i].val++;
            q[i].pos=i;
        }
        sort(q+1,q+1+m,cmp);          ///按ability降序排列

        int i=1,j=1;
        while(i<=m)
        {
            int p=a[j].pos;
            if(a[j].abi<=tabi[q[i].val])
            {
                int temp=query(1,n,1,st[q[i].val]+1,ed[q[i].val]);
                ans[q[i].pos]=temp-1;i++;
                if(i>m) break;
            }
            else if(j<n)
            {
                add(1,n,1,st[p],p);j++;
            }
        }
        for(i=1;i<=m;i++) printf("%d
",(ans[i]==0)?-1:ans[i]);
    }
    //clock_t ed=clock();
    //printf("

Time Used : %.5lf Ms.
",(double)(ed-st)/CLOCKS_PER_SEC);
    return 0;
}