hdu3950

线段树即可解决这个题。向线段树插入一个车阵的时候,只需要插入车阵的最左边的点p(姑且称这些点为黑点)即可,

len[p]记录这个车阵的长度(len数组的作用不仅仅是这些,当p这个位置为空白时,len[p]

表示从p位置开始往右的一段空白的长度)。seg结构里的lp,rp,表示区间[l, r]的黑点的位置的最小值和最小值,

sum表示黑点的总和。而minPos有点不一样,minPos表示的是长度在[l, r]之间的连续的空白段的左端点的最小值

(理解这个很重要)。

#include <iostream>
#include <stdio.h>
#include <set>
using namespace std;
const int N =  50005;
const int INF = 0X7FFFFFFF;
struct seg{
	int l, r;
	int lp, rp, minPos;
	int sum;
	bool leaf(){
		return l == r;
	}
	int mid(){
		return (l+r)>>1;
	}
}segs[N<<2];

set<int> pos[N];
int len[N];
int n, q;

template <typename T>
void getNum(T& a){
	a = 0;
	char ch;
	while(true){
		ch = getchar();
		if(ch >= '0' && ch <= '9') break;
	}
	a = ch - '0';
	while(true){
		ch = getchar();
		if(ch < '0' || ch > '9') break;
		a = a * 10 + ch - '0';
	}
}

bool input(){
	scanf("%d%d", &n, &q);
	return true;
}
inline bool check(int i){
	return i >= 1 && i <= n;
}
inline int ls(int id){
	return (id<<1)+1;
}

inline int rs(int id){
	return (id<<1)+2;
}

void build(int id, int l, int r){
	segs[id].l = l; segs[id].r = r;
	segs[id].lp = n+1; segs[id].rp = 0;
	segs[id].minPos = INF; segs[id].sum = 0;
	if(l < r){
		int mid = segs[id].mid();
		build(ls(id), l, mid);
		build(rs(id), mid+1, r);
	}else{
		pos[l].clear();
	}
}

void insBlank(int id, int p, int tlen, bool flag){
	//flag为true表示添加,为false表示删除
	//在p位置插入(删除)一段长度为tlen的空白段。
	if(segs[id].leaf()){
		if(flag){
			pos[tlen].insert(p);
			len[p] = tlen;
		}else{
			pos[tlen].erase(p);
		}
		if(pos[tlen].size() > 0){
			segs[id].minPos = *pos[tlen].begin();
		}else{
			segs[id].minPos = INF;
		}
		return ;
	}
	if(tlen <= segs[id].mid()){
		insBlank(ls(id), p, tlen, flag);
	}else{
		insBlank(rs(id), p, tlen, flag);
	}

	segs[id].minPos = min(segs[ls(id)].minPos, segs[rs(id)].minPos);
}

void insCars(int id, int p, bool flag){
	//flag为true表示添加,为false表示删除
	//在位置p插入(或删除)一个黑点
	if(segs[id].leaf()){
		if(flag){
			segs[id].sum++;
		}else{
			segs[id].sum--;
		}
		if(segs[id].sum == 0){
			segs[id].lp = n+1;
			segs[id].rp = 0;
		}else{
			segs[id].lp = segs[id].l;
			segs[id].rp = segs[id].l;
		}
		return ;
	}
	if(p <= segs[id].mid()){
		insCars(ls(id), p, flag);
	}else{
		insCars(rs(id), p, flag);
	}
	segs[id].sum = segs[ls(id)].sum + segs[rs(id)].sum;
	segs[id].lp = min(segs[ls(id)].lp, segs[rs(id)].lp);
	segs[id].rp = max(segs[ls(id)].rp, segs[rs(id)].rp);
}

int qryBlank(int id, int l, int r){  //查询长度在[l, r]之间的空白段的起点的最小值
	if(l <= segs[id].l && segs[id].r <= r){
		return segs[id].minPos;
	}
	if(segs[id].leaf()) return INF;
	int mid = segs[id].mid();
	if(r <= mid) return qryBlank(ls(id), l, r);
	else if(l > mid) return qryBlank(rs(id), l, r);
	else{
		return min(qryBlank(ls(id), l, r), qryBlank(rs(id), l, r));
	}
}
int getl(int id, int l, int r){  //获取区间[l,r]最左的黑点的位置
	if(l <= segs[id].l && segs[id].r <= r){
		return segs[id].lp;
	}
	if(segs[id].leaf()) return n+1;
	int mid = segs[id].mid();
	if(r <= mid) return getl(ls(id), l, r);
	else if(l > mid) return getl(rs(id), l, r);
	else return min(getl(ls(id), l, r), getl(rs(id), l, r));
}
int getr(int id, int l, int r){ //取区间[l,r]最右的黑点的位置
	if(l <= segs[id].l && segs[id].r <= r){
		return segs[id].rp;
	}
	if(segs[id].leaf()) return 0;
	int mid = segs[id].mid();
	if(r <= mid) return getr(ls(id), l, r);
	else if(l > mid) return getr(rs(id), l, r);
	else return max(getr(ls(id), l, r), getr(rs(id), l, r));
}

int getPos(int m, int l, int r){
	if(m > n) return INF;
	if(segs[0].sum == 0){
		return 1;
	}
	int pos;
	pos = segs[0].lp;
	if(pos - 1 >= m){
		return max(pos - r - m, 1);
	}
	pos = qryBlank(0, m, m+l+r);
	if(pos <= n){
		int tr = getl(0, pos, n);
		if(tr < 1) tr = n + 1;

		if(tr < n+1){ //右边还有车阵
			pos = max(pos, tr-r-m);
		}
		return pos;
	}
	pos = segs[0].rp + len[segs[0].rp] - 1;
	if(n - pos >= m){
		return pos+1;
	}
	return INF;
}

void ins(int p, int m, bool flag){  //flag为true表示插入一排车,为false表示离开
	int l, r;
	if(p == 1){
		l = 0;
	}else{
		l = getr(0, 1, p - 1);
		if(!check(l)) l = 0;
		else{
			l = l + len[l] - 1;
		}
	}
	if(p+m-1 == n){
		r = n+1;
	}else{
		r = getl(0, p+m, n);
		if(!check(r)) r = n+1;
	}
	if(r-l-1 > 0){
		insBlank(0, l+1, r-l-1, !flag);
	}
	if(p-l-1 > 0){
		insBlank(0, l+1, p-l-1, flag);
	}
	if(r-p-m > 0){
		insBlank(0, p+m, r-p-m, flag);
	}
}
int qrykth(int id, int k){ //查询第k个黑点的位置
	if(segs[id].leaf()) return segs[id].l;
	if(segs[ls(id)].sum >= k) return qrykth(ls(id), k);
	return qrykth(rs(id), k-segs[ls(id)].sum);
}
int cnt = 0;
void solve(){
	build(0, 1, n);
	insBlank(0, 1, n, true);
	int i, m, l, r, k;
	char op[3];
	printf("Case #%d:\n", ++cnt);
	for(i = 1; i <= q; i++){
		scanf("%s", op);
		if(op[0] == 'A'){
			getNum(m);
			getNum(l);
			getNum(r);
			int p;
			p = getPos(m, l, r);
			if(p <= n){
				printf("%d\n", p);
				ins(p, m, true);
				insCars(0, p, true);
				len[p] = m;
			}else{
				printf("-1\n");
			}
		}else{
			getNum(k);
			if(k <= segs[0].sum){
				int p = qrykth(0, k);
				ins(p, len[p], false);
				insCars(0, p, false);
			}
		}
	}
}
int main() {
	//freopen("in.txt", "r", stdin);
	int t;
	scanf("%d", &t);
	while(t--){
		input();
		solve();
	}
	return 0;
}
 

你可能感兴趣的:(线段树)