hdu 5493(线段树)

最新推荐文章于 2019-07-24 19:06:30 发布

原创最新推荐文章于 2019-07-24 19:06:30 发布 · 825 阅读

0 ·

CC 4.0 BY-SA版权

0、ACM_线段树+树状数组专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种基于线段树的算法，用于解决特定条件下队列重构的问题。该算法要求输出字典序最小且满足每人左右两侧特定数量的比他们高的个体的队列配置。通过排序和线段树更新策略，确保了算法的有效性和正确性。

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

题意：

有n个人，每个人的身高和左边(或右边)比自己高的人的个数height[i]，输出符合给出的条件且字典序最小的从左到右队列里每个人的身高。

思路：

先将每个人的身高按照从矮到高排序，当第i个人入座的时候，后面还有（n-i）个人比他高，第i个人要求左边或右边比他高的人数为k，如果n-i<k，那么说明，无法找到一个合法的序列；当我们安排第i个人入座时，只要保证他的左边（右边）还有k个空位就可以了。但是由于题意要求身高字典序最小，我们可以在左边留k个空位或者右边，（具体留左边还是右边，看第i个人坐下去时，哪个字典序更小）

采用线段树，查找当前i应该坐在哪，并更新剩余座位数。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 1e5+10,INF = 0x3f3f3f3f,MOD = 1e9+7;

typedef  pair<int,int> P;
P a[N];
int n,seat[N];
int segtree[270010]; //260010 不通过

#define  lson l,mid,now<<1
#define  rson mid+1,r,now<<1|1

void build(int l,int r,int now)
{
    segtree[now]=r-l+1;//当前区间还有多少个空座位
    if(l==r) return;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
}
void update(int k,int height,int l,int r,int now)
{
    --segtree[now];
    if (l==r){
        seat[l]=height;
        return ;
    }
    int mid=(l+r)>>1;
    if(segtree[now<<1]>k)update(k,height,lson);//左区间空席位>k
    else update(k-segtree[now<<1],height,rson);//右区间还需要k-segtree[now>>1]
}
int main()
{
    //ios::sync_with_stdio(false);
    int T,kcase=0;
    int k;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;++i)
            scanf("%d%d",&a[i].first,&a[i].second);
        sort(a+1,a+n+1);
        build(1,n,1);
        bool ok=true;
        for(int i=1;i<=n;++i)
        {
            k=a[i].second;
            if(n-i<k)//剩下的比他高的人数(n-i)<他需要比他高的人数k
            {
                ok=false;
                break;
            }
            //需要k个人在左边，把k个人安排在右边后剩下的放左边
            k = min(k, n - i - k);//☆☆☆题目要求字典序小，故尽量靠左放
            update(k,a[i].first,1,n,1);
        }
      printf("Case #%d: ", ++kcase);
      if(!ok)puts("impossible");
      else for(int i=1;i<=n;++i)printf("%d%c",seat[i]," \n"[i==n]);
    }
    return 0;
}