BZOJ1208 [HNOI2004]宠物收养所

[Problem]

最近，阿Q开了一间宠物收养所。收养所提供两种服务：收养被主人遗弃的宠物和让新的主人领养这些宠物。每个领养者都希望领养到自己满意的宠物，阿Q根据领养者的要求通过他自己发明的一个特殊的公式，得出该领养者希望领养的宠物的特点值a（a是一个正整数，a<2^31），而他也给每个处在收养所的宠物一个特点值。这样他就能够很方便的处理整个领养宠物的过程了，宠物收养所总是会有两种情况发生：被遗弃的宠物过多或者是想要收养宠物的人太多，而宠物太少。 1． 被遗弃的宠物过多时，假若到来一个领养者，这个领养者希望领养的宠物的特点值为a，那么它将会领养一只目前未被领养的宠物中特点值最接近a的一只宠物。（任何两只宠物的特点值都不可能是相同的，任何两个领养者的希望领养宠物的特点值也不可能是一样的）如果有两只满足要求的宠物，即存在两只宠物他们的特点值分别为a-b和a+b，那么领养者将会领养特点值为a-b的那只宠物。 2． 收养宠物的人过多，假若到来一只被收养的宠物，那么哪个领养者能够领养它呢？能够领养它的领养者，是那个希望被领养宠物的特点值最接近该宠物特点值的领养者，如果该宠物的特点值为a，存在两个领养者他们希望领养宠物的特点值分别为a-b和a+b，那么特点值为a-b的那个领养者将成功领养该宠物。 一个领养者领养了一个特点值为a的宠物，而它本身希望领养的宠物的特点值为b，那么这个领养者的不满意程度为abs(a-b)。【任务描述】你得到了一年当中，领养者和被收养宠物到来收养所的情况，希望你计算所有收养了宠物的领养者的不满意程度的总和。这一年初始时，收养所里面既没有宠物，也没有领养者。

[Solution]

What can be more pleasing than finding an undone easy problem?

Just a practice of stl set.

[Code]

#include <cstdio>
#include <algorithm>
#include <set>
#include <memory.h>

using namespace std;

typedef multiset <int> bst;
typedef bst :: iterator bstiter;

const int maxn = 100010;
const int mod =	1000000;

int n, s;
bst c, d;

int getD(int v) {
	bstiter a = d. lower_bound(v);
	if (a == d. end())
		return *d. rbegin();
	bstiter b = a;
	b --;
	if (b == d. end())
		return *a;
	if (abs(*b - v) <= abs(*a - v))
		return *b;
	else
		return *a;
}

int getC(int v) {
	bstiter a = c. lower_bound(v);
	if (a == c. end())
		return *c. rbegin();
	bstiter b = a;
	b --;
	if (b == c. end())
		return *a;
	if (abs(*b - v) <= abs(*a - v))
		return *b;
	else
		return *a;
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
#endif

	scanf("%d", &n);
	s = 0;
	for (int i = 0; i < n; i ++) {
		int o, v;
		scanf("%d%d", &o, &v);
		if (o == 0) {
			if (d. empty())
				c. insert(v);
			else {
				int r = getD(v);
				d. erase(d. find(r));
				(s += abs(r - v)) %= mod;
			}
		}
		else {
			if (c. empty())
				d. insert(v);
			else {
				int r = getC(v);
				c. erase(c. find(r));
				(s += abs(r - v)) %= mod;
			}
		}
	}
	printf("%d\n", s);
}