[2017纪中11-6]奇怪的队列 树状数组+二分/线段树

题面
考虑先按高度排序,那么第i位置的人一共有n-i个人比他高,为了方便,这个人报的数字t=min(t,n-i-t)。
那么从小到大把这些人放进序列里。每次放在当前序列t的位置上,然后把这个位置删除即可。
可以用树状数组+二分搞。复杂度O(nlog^2n)。
线段树正解
nlog^2n代码:

#include
#include
#include
using namespace std;
const int maxn=100010;
int n,c[maxn],p[maxn];
struct node
{
    int h,t;
}a[maxn];
bool cmp(node a,node b)
{
    return a.hvoid add(int x,int d){for(;x<=n;x+=(x&(-x))) c[x]+=d;}
int qry(int x) {int r=0;for(;x>0;x-=(x&(-x))) r+=c[x]; return r;}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&a[i].h,&a[i].t);
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++)
    {
        if(a[i].t>(n-i)/2) a[i].t=n-i-a[i].t;
        if(a[i].t<0) {puts("impossible"); return 0;}
        if(i!=1)add(i,-1);
    }
    for(int i=1;i<=n;i++)
    {
        int l=1,r=n;
        while(lint mid=(l+r)>>1;
            if(qry(mid)<=-a[i].t) r=mid;
            else l=mid+1;
        }
        add(l,1);
        p[l]=a[i].h;
    }
    for(int i=1;i<=n;i++)
        printf("%d ",p[i]);
    return 0;
}

你可能感兴趣的:(二分,树状数组,线段树)