1819 郁闷的出纳员之树状数组解法

这个题目最近很火呵，FallingFlowers的空间有SBT的解法，我又从icecream那学来了线段树的解法。最近在网上发现树状数组有查询第k大的功能，想到用在这个题目上，于是有了树状数组解法，试了一下，速度很快，0.45s，树状数组的效率还真是BT啊。

这个题目的要求就是插入元素，删除元素，查询第k大。插入就不用说了，删除的时候用一个堆保存插入过的元素，然后从前到后依次删除，直到堆顶元素不小于给定的下界即可。查询第k大的代码见下：（有些地方我自己修改了一下，感觉网上的版本有些不太对）

int findK(int k)
{
    int pos = 0, now = 0;

     for (int i = (int)log2(MAX-1); i >= 0; i--) //最好写成MAX-1，防止MAX正好是2的幂次造成溢出
     {
         pos += 1 << i;
         if (now + c[pos] < k)   //此处必须是小于，否则如果是一个元素会出错
             now += c[pos];
         else
            pos -= 1 << i;
     }

    return pos + 1;
}

这样三个功能就都实现了，其余要注意的就是由于要维护一个偏移量（统一加减工资），因此在插入元素的时候要加上一个值，防止插入的元素是负的。

主程序见下：

    #include <cstdio>
    #include <cmath>
    #include <queue>
    #define MAX 200010
    #define INC 10001
    using namespace std;

    int n, delta, v, down, cnt;
    char op[4];

    while (scanf("%d %d", &n, &down) == 2)
    {
        delta = cnt = 0;
        priority_queue<int, vector<int>, cmp> q;
        memset(c, 0, sizeof(c));
        for (int i = 0; i < n; i++)
        {
            scanf("%s %d", op, &v);
            if (op[0] == 'F')
            {
                if (v > q.size())
                    printf("-1\n");
                else
                    printf("%d\n", findK(q.size() - v + 1) + delta - INC);
            }
            else if (op[0] == 'I')
            {
                if (v >= down)
                {
                    modify(v - delta + INC, 1);
                    q.push(v - delta);
                }
            }
            else if (op[0] == 'A')
                delta += v;
            else
            {
                delta -= v;
                while (q.size())
                {
                    if (q.top() + delta >= down)
                        break;
                    modify(q.top() + INC, -1);
                    q.pop();
                    cnt++;
                }
            }
        }
        printf("%d\n", cnt);
    }

明天有空再写下线段树的版本。树状数组就是应用范围太窄，要不然真就是Perfect了。

P.S.用树状数组求第k大的代码有些问题，中间运算会溢出，安全的方法是在开c数组上界的时候要开到大于给定的上界的最小2的方幂那么大。