HDU 5372 - Segment Game(树状数组)

题目：

http://acm.hdu.edu.cn/showproblem.php?pid=5372

题意：

在水平坐标轴上，0操作表示画线段（b，b+i），1操作表示将线段i删去，输出每次增加的线段包含的线段的数量。

思路：

由于b有10^9，所以需要离散化，将所有的坐标存入num数组中，排序去从。在num数组中二分查找每次询问的左右坐标，得到的下标即为离散后的新坐标。

维护两个树状数组，一个表示左端点，一个表示右端点，每次S【r】- S【l-1】即为区间（l，r）的完整线段数量。

做线段删除操作时，两个树状数组都在左端点位置向上减1 即可。

AC.

 #include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#pragma comment (linker,"/STACK:1024000000,1024000000")

using namespace std;
const int maxn = 4e5+5;
int p[maxn], a[maxn], b[maxn], num[maxn];
int cnt, s[2][maxn];

int bit(int x)
{
return x & (-x);
}
void add(int x, int d, int o)
{
while(x <= cnt) {
s[o][x] += d;
x += bit(x);
}
}
int sum(int x, int o)
{
int ret = 0;
while(x > 0) {
ret += s[o][x];
x -= bit(x);
}
return ret;
}

int main()
{
// freopen("in", "r", stdin);
int n, ca = 0;
while(~scanf("%d", &n)) {
cnt = 0;
int c = 0;
for(int i = 1; i <= n; ++i) {
scanf("%d %d", &p[i], &a[i]);
if(!p[i]) {
b[i] = a[i]+ (++c);
num[cnt++] = a[i];
num[cnt++] = b[i];
}
}

sort(num, num+cnt);
cnt = unique(num, num+cnt) - num;
memset(s, 0, sizeof(s));

int ad = 1;
printf("Case #%d:\n", ++ca);
for(int i = 1; i <= n; ++i) {
if(p[i] == 0) {
int aa = lower_bound(num, num+cnt, a[i]) - num + 1;
int bb = lower_bound(num, num+cnt, b[i]) - num + 1;

printf("%d\n", sum(bb, 1) - sum(aa-1, 0));

a[ad] = aa;
b[ad++] = bb;

add(aa, 1, 0);
add(bb, 1, 1);
}
else {
add(a[a[i]], -1, 0);
add(b[a[i]], -1, 1);
}
}

}
return 0;
}