【ACWing】260. 买票

题目地址：

https://www.acwing.com/problem/content/description/262/

达达在买回家的火车票，因为正值春运，售票处排起了长队。因为晚上室内光线很暗，所以很多人趁机插队。现在给每个人赋予一个整数作为编号，告诉你每一个排队的人的编号，和他进入队列时的具体位置。请你确定最终的队列顺序。

输入格式：
输入可能包含多组测试用例。对于每组测试用例，第一行包含整数$N$，表示排队的总人数。接下来$N$行，每行两个整数$P_i,V_i$，第$i$行数据表示第$i$个人进入队列时的位置以及他的个人编号。一个人的$P_i$值具体表示为当该人员进入队列时，他前面的人数。例如，如果一个人插到了队首，则其$P_i$值为$0$，如果插到了第三个位置（第二个人后面），则其$P_i$值为$2$。

输出格式：
每个测试用例，输出一行包含$N$个整数（表示每个人的编号）的结果，表示最终的人员队列顺序。每个结果占一行，同行数据之间空格隔开。

数据范围：
$1\le N\le 200000$
$0\le V_i\le 32767$
$0\le P_i\le i-1$

思路和https://blog.csdn.net/qq_46105170/article/details/121484545是一样的。先开一个维护长度$N$的$01$序列的树状数组，一开始它维护的序列数字都是$1$。我们可以倒着遍历这$N$个人，想象一下时光倒流所发生的场景，如果最后一个到达的人前面有$k$个人，那他的位置就是$k+1$；接着将这个人删掉（对应的是将树状数组第$k+1$个位置的数减$1$），看倒数第二个到达的人前面有多少人，假设有$k^{\prime }$个人，那他的位置就是要找$01$序列里前缀和从左到右第一次大于$k^{\prime }$的位置，然后再删掉，找倒数第三个人，以此类推。所以说，要找某个人的位置，实际上就是要找前缀和第一次大于某个数的位置，这可以用树状数组 + 二分来做，每次寻找时间复杂度$O({\log }^{2}n)$，总复杂度$O(n{\log }^{2}n)$。代码如下：

#include <iostream>
using namespace std;

const int N = 2e5 + 5;
int tr[N], p[N], v[N], res[N];
int n;

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

void add(int k, int x) {
    for (int i = k; i <= n; i += lowbit(i)) tr[i] += x;
}

int sum(int k) {
    int res = 0;
    for (int i = k; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main() {
    while (cin >> n) {
        for (int i = 1; i <= n; i++) {
            scanf("%d%d", &p[i], &v[i]);
            tr[i] = lowbit(i);
        }

        for (int i = n; i; i--) {
            int l = 1, r = n;
            while (l < r) {
                int mid = l + r >> 1;
                if (sum(mid) > p[i]) r = mid;
                else l = mid + 1;
            }

            res[l] = i;
            add(l, -1);
        }

        for (int i = 1; i <= n; i++) printf("%d ", v[res[i]]);
        puts("");
    }

    return 0;
}

时间复杂度$O(n{\log }^{2}n)$，空间$O(n)$。