hdu 4366 Successor(线段树）

最新推荐文章于 2019-08-13 12:42:01 发布

AC_way

最新推荐文章于 2019-08-13 12:42:01 发布

阅读量558

点赞数

分类专栏： ACM 线段树文章标签： algorithm 数据结构

ACM 线段树专栏收录该内容

7 篇文章

订阅专栏

本文探讨了一个涉及点与点间上下级关系的问题，通过将树状结构转化为线状结构，实现了一种高效的算法来解决如何选择最优节点进行删除的操作。此算法包括树的遍历、线段树的构建与更新，以及忠诚度与能力值的匹配策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

转载自http://blog.youkuaiyun.com/weiguang_123/article/details/8143916

题意：
给你 n个点。他们有上下级关系，一个点只有一个上级，一个上级可以有多个下级，每个点有两个属性，能力值、忠诚度（每个节点的忠诚度不同），求我们要删除一个节点，则我们需从其下级中选出一个节点，其能力值比该节点要高，且忠诚度是（比其能力高的下级节点中的）最高
题解：
首先将树状结构，转化为线状结构，
可以遍历一遍将树上每个点标记为一维区间上的某个点，且在同一棵子树内的点是连续的一段。
然后，将所有点按能力从大到小排序，能力相同的编号小的排在前面，然后扫描一遍，扫描时维护一颗线段树，
（我们先插入线段树的是，比其能力值大的，我们只要从这些里面找到，忠诚度最高的就可以了）
先查找该点为根节点的子树内的最优值，然后插入该点的忠诚度。

c++提交会爆栈，手动扩栈

#pragma comment(linker, "/STACK:10240000000000,10240000000000")
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int maxn=100000+1000;
map<int,int> ma;
vector<int> son[maxn];
int n,m,num,ans[maxn],l[maxn],r[maxn];
struct staf
{
    int abt;
    int loy;
    int id;
}staff[maxn];
struct node
{
    int l;
    int r;
    int maxl;
}tree[maxn*2];
bool cmp(staf a,staf b)
{
    return a.abt>b.abt;
}
void dfs(int u)//将树形转化为数组
{
    l[u]=num++;
    for(int i=0;i<son[u].size();i++)
    {
        dfs(son[u][i]);
    }
    r[u]=num;
}
void build(int rt,int l,int r)
{
    tree[rt].l=l;
    tree[rt].r=r;
    tree[rt].maxl=-1;
    if(l==r)
    return;
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
}
int get(int x,int l,int r)
{
    if(r<l)
    return -1;
    if(l==tree[x].l&&r==tree[x].r)
    {
        return tree[x].maxl;
    }
    int mid=(tree[x].l+tree[x].r)>>1;
    if(r<=mid)
    {
       return get(x<<1,l,r);
    }
    else
    {
        if(l>mid)
        {
           return get(x<<1|1,l,r);
        }
        else
        {
            return max(get(x<<1,l,mid),get(x<<1|1,mid+1,r));
        }
    }
}
void insert(int x,int rt,int v)//单点插入更新
{
    if(tree[x].l==tree[x].r)
    {
        tree[x].maxl=v;
        return;
    }
    int mid=(tree[x].l+tree[x].r)>>1;
    if(rt<=mid)
    {
        insert(x<<1,rt,v);
    }
    else
    {
        insert(x<<1|1,rt,v);
    }
    tree[x].maxl=max(tree[x<<1].maxl,tree[x<<1|1].maxl);
}
void init()
{
    ma.clear();
    for(int i=0;i<=n;i++)
    son[i].clear();
    num=0;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        int fa;
        init();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&fa,&staff[i].loy,&staff[i].abt);
            son[fa].push_back(i);
            staff[i].id=i;
            ma[staff[i].loy]=i;
        }
        dfs(0);
        sort(staff+1,staff+n,cmp);
        build(1,0,num-1);
        int j;
        for(int i=1;i<n;i=j)
        {
            j=i;
           while(j<n&&staff[i].abt==staff[j].abt)
           {
               int id=staff[j].id;
               int loy=get(1,l[id]+1,r[id]-1);
               if(loy != -1) ans[id] = ma[loy] ;
                else ans[id] = -1;
                ++j;
           }
           j = i ;
            while(j < n && staff[i].abt == staff[j].abt )
            {
                int id =  staff[j].id;
                insert(1,l[id],staff[j].loy);
                ++j;
            }
        }
        int a;
        while(m--)
        {
            scanf("%d",&a);
            printf("%d\n",ans[a]) ;
        }
    }
    return 0;
}