G - Queue HDU - 5493 线段树+二分

G - Queue

HDU - 5493

题目大意：给你n个人的身高和这个人前面或者后面有多少个比他高的人，让你还原这个序列，按字典序输出。

题解：

首先按高度排序。

设每个人在其前面有k个人，设比这个人高有x个人，所以k=min(k,x-k)，求出每一个人在他前面比他高的人的数量。

然后用线段树来维护一段区间的空位，因为对于每一个人来说比他高的人肯定都是出现在他后面，所以这个就说明它前面要空出几个空位。

但是线段树可以维护一段区间的空位，没办法准确找出我需要空位为一个准确的数的位置。

这个就可以二分答案。

要注意几个细节，如果用线段树的话，这个 l 不要从1开始，不然会 T，

然后就是二分这个地方要注意写法，我之前一种写法错了。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 1e5 + ;
int ans[maxn];
int n;
struct node {
    int h, k;
    node(int h = , int k = ) :h(h), k(k) {}
}ex[maxn];

bool cmp(node a, node b) {
    return a.h < b.h;
}
int sum[maxn * ];
void push_up(int id) {
    sum[id] = sum[id << ] + sum[id <<  | ];
}

void build(int id, int l, int r) {
    sum[id] = ;
    if (l == r) return;
    int mid = (l + r) >> ;
    build(id << , l, mid);
    build(id <<  | , mid + , r);
}

void update(int id, int l, int r, int pos) {
    if (l == r) {
        sum[id] = ;
        return;
    }
    int mid = (l + r) >> ;
    if (pos <= mid) update(id << , l, mid, pos);
    else update(id <<  | , mid + , r, pos);
    push_up(id);
}

int query(int id, int l, int r, int x, int y) {
    if (x <= l && y >= r) return r - l +  - sum[id];
    int mid = (l + r) >> ;
    int ans = ;
    if (x <= mid) ans += query(id << , l, mid, x, y);
    if (y > mid) ans += query(id <<  | , mid + , r, x, y);
    return ans;
}

int main() {
    int t, cnt = ;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = ; i <= n; i++) {
            int h, k;
            scanf("%d%d", &h, &k);
            ex[i] = node(h, k);
        }
        int flag = ;
        sort(ex + , ex +  + n, cmp);
        for (int i = ; i <= n; i++) {
            if (n - i - ex[i].k < ) flag = ;
            ex[i].k = min(ex[i].k, n - i - ex[i].k);
        }
        if (flag) {
            printf("Case #%d: impossible\n", ++cnt);
            continue;
        }
        build(, , n);
        for (int i = ; i <= n; i++) {
            int l = ex[i].k + , r = n;
            while (l <= r) {
                int mid = (l + r) >> ;
                int res = query(, , n, , mid);
                if (res > ex[i].k) r = mid - ;
                else l = mid + ;
            }
            ans[r + ] = ex[i].h;
            update(, , n, r + );
        }
        printf("Case #%d:", ++cnt);
        for (int i = ; i <= n; i++) printf(" %d", ans[i]);
        printf("\n");
    }
    return ;
}

线段树

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <stack>
#include <vector>
#include <algorithm>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 1e5 + ;
int ans[maxn];
int n;
struct node {
    int h, k;
    node(int h = , int k = ) :h(h), k(k) {}
}ex[maxn];

bool cmp(node a, node b) {
    return a.h < b.h;
}
int sum[maxn];

int lowbit(int x)
{
    return x & (-x);
}

void update(int i,int k)
{
    while(i<n)
    {
        sum[i] += k;
        i += lowbit(i);
    }
}

int getsum(int x)
{
    int res = ;
    while(x>)
    {
        res += sum[x];
        x -= lowbit(x);
    }
    return res;
}

int main() {
    int t, cnt = ;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = ; i <= n; i++) {
            sum[i] = ;
            int h, k;
            scanf("%d%d", &h, &k);
            ex[i] = node(h, k);
        }
        int flag = ;
        sort(ex + , ex +  + n, cmp);
        for (int i = ; i <= n; i++) {
            if (n - i - ex[i].k < ) {
                flag = ;
                break;
            }
            ex[i].k = min(ex[i].k, n - i - ex[i].k);
        }
        if (flag) {
            printf("Case #%d: impossible\n", ++cnt);
            continue;
        }
        for (int i = ; i <= n; i++) {
            int l = ex[i].k + , r = n;
            int num = ex[i].k + ;
            while (l <= r) {
                int mid = (l + r) >> ;
                int res = mid - getsum(mid), e = getsum(mid) - getsum(mid - );
                // if (res == num && e == 0) {
                //     ans[mid] = ex[i].h;
                //     update(mid, 1);
                //     flag1 = 1;
                //     break;
                // }
                if (res >= num) r = mid - ;
                else l = mid + ;
            }
            ans[r + ] = ex[i].h;
            update(r + , );
        }
        printf("Case #%d:", ++cnt);
        for (int i = ; i <= n; i++) printf(" %d", ans[i]);
        printf("\n");
    }
    return ;
}

树状数组