UVA 1391 - Astronauts（2-SET）

UVA 1391 - Astronauts

題目鏈接

題意：給定一些宇航員，年齡小於平均數能做A和C，大於等於能做B和C，現在知道一些宇航員互相憎恨，不能讓他們做同一個任務，問一直種安排方法滿足條件

思路：2set問題，如果兩種宇航員類型相同，就兩個宇航員做不一樣，加一條真或真，和假或假的邊，如果類型不同，就加一條真或真的邊

代碼：

#include 
#include 
#include 
#include 
#include 
using namespace std;

const int MAXNODE = 100005;

struct TwoSet {
	int n;
	vector g[MAXNODE * 2];
	bool mark[MAXNODE * 2];
	int S[MAXNODE * 2], sn;

	void init(int tot) {
		n = tot * 2;
		for (int i = 0; i < n; i += 2) {
			g[i].clear();
			g[i^1].clear();
		}
		memset(mark, false, sizeof(mark));
	}

	void add_Edge(int u, int uval, int v, int vval) {
		u = u * 2 + uval;
		v = v * 2 + vval;
		g[u^1].push_back(v);
		g[v^1].push_back(u);
	}

	bool dfs(int u) {
		if (mark[u^1]) return false;
		if (mark[u]) return true;
		mark[u] = true;
		S[sn++] = u;
		for (int i = 0; i < g[u].size(); i++) {
			int v = g[u][i];
			if (!dfs(v)) return false;
		}
		return true;
	}

	bool solve() {
		for (int i = 0; i < n; i += 2) {
			if (!mark[i] && !mark[i + 1]) {
				sn = 0;
				if (!dfs(i)){
					for (int j = 0; j < sn; j++)
						mark[S[j]] = false;
					sn = 0;
					if (!dfs(i + 1)) return false;
				}
			}
		}
		return true;
	}
} gao;

const int N = 100005;

int n, m, age[N], sum;
vector g[N];

int main() {
	while (~scanf("%d%d", &n, &m) && n) {
		sum = 0;
		gao.init(n);
		for (int i = 0; i < n; i++) {
			scanf("%d", &age[i]);
			g[i].clear();
			sum += age[i];
		}
		for (int i = 0; i < n; i++) {
			if (age[i] * n < sum)
				age[i] = 0;
			else
				age[i] = 1;
		}
		int u, v;
		while (m--) {
			scanf("%d%d", &u, &v);
			u--; v--;
			g[u].push_back(v);
		}
		for (int u = 0; u < n; u++) {
			for (int j = 0; j < g[u].size(); j++) {
				int v = g[u][j];
				if (age[u]^age[v])
					gao.add_Edge(u, 1, v, 1);
				else {
					gao.add_Edge(u, 0, v, 0);
					gao.add_Edge(u, 1, v, 1);
				}
			}
		}
		if (gao.solve()) {
			for (int i = 0; i < n; i++) {
				if (age[i] && gao.mark[i * 2 + 1]) printf("A\n");
				else if (age[i] == 0 && gao.mark[i * 2 + 1]) printf("B\n");
				else printf("C\n");
			}
		} else printf("No solution.\n");
	}
	return 0;
}