HDU 5372 Segment Game(2015 Multi-University Training Contest 7 1004)

题目链接：

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

题意：

给出n个操作，每个操作有a, b两个数。a=0表示在数轴上放一条端点分别是b和c的线段，c=b+(这次操作是第几次放置线段的操作)。a=1表示删掉第b次放置线段操作所放置的线段。每次a=0时要求输出要插入的线段完全覆盖了多少条线段。

思路：

由于数据范围比较大，先读入所有操作，将所有线段的端点离散化。之后用树状数组依次处理操作，每次要输出的答案为（左端点>=当前线段左端点的线段个数）-（右端点>该线段右端点的个数）。

好久没写离散化，mark一下

code：

#include <iostream>
#include <cstdio>
#include <vector>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;

//离散化
class Discr{
public:
	int data[400000 + 10];
	int cnt;

	Discr(){
		cnt = 0;
	}

	//向数字集合里加入一个数
	void add(int a){
		data[cnt++] = a;
	}

	//排序并去重
	void sort_and_unique() {
		sort(data, data + cnt);
		int i = 0, j = 0;
		for (; i < cnt; ){
			data[j] = data[i];
			while (i < cnt && data[j] == data[i])i++;
			j++;
		}
		cnt = j;
	}

	//二分查找某个数离散化之后是什么数字
	//从1开始
	int get_id(int num){
		return lower_bound(data, data + cnt, num) - data + 1;
	}
};
Discr dis;

//树状数组
class BinaryIndexedTree{
public:
	int in[262150];
	static int MAX_SIZE;

	void init(){
		memset(in, 0, sizeof(in));
	}

	inline int lowbit(int x){
		return x&(-x);
	}

	// end >= 1
	int sum(int end){
		int sum = 0;
		while (end){
			sum += in[end];
			end -= lowbit(end);
		}
		return sum;
	}

	// pos >= 1
	void add(int pos, int num){
		while (pos <= MAX_SIZE){
			in[pos] += num;
			pos += lowbit(pos);
		}
	}
}bt1, bt2;

int BinaryIndexedTree::MAX_SIZE = 262144;

struct Query{
	int type;
	int b, c;
};
Query query[200010];
int addId[200010];
int n;

int main(){
	int __ = 0;
	while (scanf("%d", &n) != EOF){
		dis.cnt = 0;
		bt1.init(); bt2.init();
		for (int i = 1, j = 1; i <= n; i++){
			int a, b;
			scanf("%d%d", &query[i].type, &query[i].b);
			if (query[i].type == 0){
				addId[j++] = i;
				query[i].c = query[i].b + j;
				dis.add(query[i].b);
				dis.add(query[i].c);
			}
		}
		dis.sort_and_unique();
		printf("Case #%d:\n", ++__);
		for (int i = 1; i <= n; i++){
			if (query[i].type == 0){
				int l = dis.get_id(query[i].b);
				int r = dis.get_id(query[i].c);
				printf("%d\n", bt2.sum(r) - bt1.sum(l - 1));
				bt1.add(l, 1);
				bt2.add(r, 1);
			}
			else{
				int id = addId[query[i].b];
				int l = dis.get_id(query[id].b);
				int r = dis.get_id(query[id].c);
				bt1.add(l, -1);
				bt2.add(r, -1);
			}
		}
	}
	return 0;
}