[LeetCode]406 Queue Reconstruction by Height（BIT，二分）

题目链接：https://leetcode.com/problems/queue-reconstruction-by-height/?tab=Description

题意：给一个队列，每个人的身高以及他们前面的比他高的人的数量。求重构这个队列。

很简单，以前意淫过一道类似的题，不过是逆序数。这道题明显条件多了一点，不过不影响。

按照身高从小到大排序，然后在此基础上按照逆序数从大到小排序。用树状数组维护[1,pos]的总的空白位置数量，每次插入的时候从头到尾插，二分找到最左边的满足条件的位置后，更新空白数量-1即可。

甚至感觉像在写模拟。

class Solution {
public:
    int n, bit[1100];
    int lowbit(int x) {
        return x & (-x);
    }
    void update(int i, int x) {
        while(i <= n) {
            bit[i] += x;
            i += lowbit(i);
        }
    }
    int sum(int i) {
        int ret = 0;
        while(i) {
            ret += bit[i];
            i -= lowbit(i);
        }
        return ret;
    }
    int lb(int val) {
        int lo = 1, hi = n;
        while(lo <= hi) {
            int mid = (lo + hi) >> 1;
            int x = sum(mid);
            if(x >= val) hi = mid - 1;
            else lo = mid + 1;
        }
        return lo;
    }
    static bool cmp(const pair<int, int>& a, const pair<int, int>& b) {
        if(a.first != b.first) return a.first < b.first;
        return a.second > b.second;
    }
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        pair<int, int> ret[1100];
        memset(bit, 0, sizeof(bit));
        n = people.size();
        for(int i = 2; i <= n; i++) update(i, 1);
        sort(people.begin(), people.end(), cmp);
        for(int i = 1; i <= n; i++) {
            int pos = lb(people[i-1].second+1);
            update(pos, -1);
            ret[pos-1].first = people[i-1].first;
            ret[pos-1].second = people[i-1].second;
        }
        people.clear();
        for(int i = 1; i <= n; i++) people.push_back(ret[i]);
        return people;
    }
};