这个题目最近很火呵,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的方幂那么大。