【KickStart】Round H 2019: H-index

起因：本来说好打游戏的人咕咕咕了

题目链接：https://codingcompetitions.withgoogle.com/kickstart/round/0000000000050edd/00000000001a274e

一直想多刷题来着，但是总犯懒。今天也只刷了这一道，典型的树状数组题（之前还不太会，现学现用吧……）。简单分析，题目数值范围比较大，需要快速求解大于等于H的数值有多少个，数值有大小范围，可以用一个计数数组存下来，然后树状数组求区间值就可以了。

树状数组简单理解，就是重用区间和。计算当前下标的二进制，有N个0，就往前（包括自己）数2^N个数加和。通过x & (-x)的方式消除二进制最后一个1，然后在新的下标下继续求和。

#include "bits/stdc++.h"
#include <vector>

using namespace std;

int T, N, A[100010], tmp, res, maxVal, l, r, m;

// 看的别人的树状数组模板，这个class可以不在初始化的时候接受arr的
// 为了以后方便复用还是留着吧（
class BinaryIndexedTree {
public:
    vector<int> bitArr;

    BinaryIndexedTree(int arr[], int l = 0) {
        bitArr.clear();
        bitArr.resize(l + 1, 0);
        for (int i = 0; i < l; ++i) 
            bitArr[i + 1] = arr[i];
        int j;
        for (int i = 1; i <= l; ++i) {
            j = i + (i & -i);
            if (j <= l) 
                bitArr[j] += bitArr[i];
        }
    }

    void reset(int arr[], int l = 0) {
        bitArr.clear();
        bitArr.resize(l + 1, 0);
        for (int i = 0; i < l; ++i)
            bitArr[i + 1] = arr[i];
        int j;
        for (int i = 1; i <= l; ++i) {
            j = i + (i & -i);
            if (j <= l)
                bitArr[j] += bitArr[i];
        }
    }

    void update(int idx, int val) {
        ++idx;
        while (idx < bitArr.size()) {
            bitArr[idx] += val;
            idx += idx & (-idx);
        }
    }

    int prefixSum(int idx) {
        ++idx;
        int res = 0;
        while (idx > 0) {
            res += bitArr[idx];
            idx -= idx & (-idx);
        }
        return res;
    }

    int rangeSum(int withoutLeft, int withRight) {
        return prefixSum(withRight) - prefixSum(withoutLeft);
    }
};


int main() {
    scanf("%d", &T);
    BinaryIndexedTree * binIndTree = new BinaryIndexedTree(A);
    for (int _t = 1; _t <= T; ++_t) {
        scanf("%d", &N);
        printf("Case #%d:", _t);
        res = 0;
        maxVal = 1;

        binIndTree->reset(A, 100005);
        for (int i = 1; i <= N; ++i) {
            scanf("%d", &tmp);
            binIndTree->update(tmp, 1);
            maxVal = max(tmp, maxVal);

            l = res, r = maxVal;
            while (l <= r) {
                m = (l + r) >> 1;
                if (m <= binIndTree->rangeSum(m - 1, maxVal)) {
                    res = m;
                    l = m + 1;
                }
                else {
                    r = m - 1;
                }
            }
            printf(" %d", res);
        }
        printf("
");
    }
    return 0;
}