uva 11134 - Fabled Rooks(问题转换+优先队列)

题目链接:uva 11134 - Fabled Rooks



题目大意:给出n,表示要在n*n的矩阵上放置n个车,并且保证第i辆车在第i个区间上,每个区间给出左上角和右小角的坐标。另要求任意两个车之间不能互相攻击。


解题思路:因为要保证说每两个车之间不能互相攻击,那么即任意行列都不能摆放两个以上的车,转而言之可以看成是将每一行或列分配给每辆车。如果行和列和起来考虑的话复杂度太高了,但是行和列的分配又互相不影响,所以可以分开讨论。


即对于一个区间[xl,xr],要分配一个x给它,做法和uva 1422一样。


#include <stdio.h>
#include <string.h>
#include <queue>
#include <algorithm>

using namespace std;

const int N = 5005;

struct state {
	int l, r, id;
	friend bool operator < (const state a, const state b) {
		return a.r > b.r;
	}
}x[N], y[N], ans[N];
int n;

bool cmp(const state& a, const state& b) {
	return a.l < b.l;
}

void init() {
	for (int i = 0; i < n; i++) {
		scanf("%d%d%d%d", &x[i].l, &y[i].l, &x[i].r, &y[i].r);
		x[i].id = y[i].id = i;
	}
	sort(x, x + n, cmp);
	sort(y, y + n, cmp);
}

bool solve() {
	priority_queue<state> q;
	state c;

	int p = 0;
	for (int i = 0; i < n; i++) {
		while(p < n) {
			if (x[p].l <= i + 1) q.push(x[p]);
			else break;
			p++;
		}

		if (q.empty()) return false;

		c = q.top(); q.pop();
		if (c.r < i + 1) return false;
		ans[c.id].l = i + 1;
	}
	
	p = 0;
	for (int i = 0; i < n; i++) {
		while(p < n) {
			if (y[p].l <= i + 1) q.push(y[p]);
			else break;
			p++;
		}

		if (q.empty()) return false;

		c = q.top(); q.pop();
		if (c.r < i + 1) return false;
		ans[c.id].r = i + 1;
	}

	for (int i = 0; i < n; i++) printf("%d %d\n", ans[i].l, ans[i].r);
	return true;
}

int main () {
	while (scanf("%d", &n) == 1 && n) {
		init();
		if (solve() == false) printf("IMPOSSIBLE\n");
	}
	return 0;
}


你可能感兴趣的:(uva 11134 - Fabled Rooks(问题转换+优先队列))