LA 3713 Astronauts(2-SAT)

题意：有A,B,C三个任务要分配给n个宇航员，其中每个宇航员好分配一个任务，所有宇航员的平均年龄是X,大于等于X的才能做A任务，小于X的才能做B任务，有m对宇航员互相讨厌不能分配到一个任务，求一个任务分配方案。

分析：依然是2-SAT模型，设xi表示是否不做C任务，若宇航员i和j互相讨厌且都大于X，则xi != xj,若i和j的年龄段不同，则xi or xj。

#include <queue>    
#include <vector>    
#include <cstdio>    
#include <utility>    
#include <cstring>    
#include <iostream>    
#include <algorithm>    
#define INF 0x3f3f3f3f
#define MAXN 200005
using namespace std;
int n,m;
vector <int> G[MAXN];
bool mark[MAXN];
int S[MAXN],num[MAXN],age[MAXN],c;
void init()
{
	for(int i = 0;i < 2*n;i++) G[i].clear();
	memset(mark,0,sizeof(mark));
}
void add_clause(int x,int xval,int y,int yval)
{
	x = x * 2 + xval;
	y = y * 2 + yval;
	G[y].push_back(x^1);
	G[x].push_back(y^1); 
}
bool dfs(int x)
{
	if(mark[x^1]) return false;
	if(mark[x]) return true;
	mark[x] = true;
	S[c++] = x;
	for(int v : G[x])
	 if(!dfs(v)) return false;
	return true; 
}
bool jud()
{
	for(int i = 0;i < 2*n;i += 2)
	 if(!mark[i] && !mark[i+1])
	 {
	 	c = 0;
	 	if(!dfs(i))
	 	{
	 		while(c > 0) mark[S[--c]] = false;
	 		if(!dfs(i+1)) return false;
		 }
	 }
	 return true;
}
int main()
{
	while(~scanf("%d%d",&n,&m) && n+m)
	{
		double tot = 0;
		init();
		for(int i = 0;i < n;i++)
		{
			scanf("%d",&age[i]);
			tot = tot + age[i];
		}
		tot /= n;
		for(int i = 0;i < n;i++)
		 if(age[i] >= tot) num[i] = 1;
		 else num[i] = 2;
		for(int i = 1;i <= m;i++)
		{
			int x,y;
			scanf("%d%d",&x,&y);
			if(num[--x] == num[--y])
			{
				add_clause(x,1,y,1);
				add_clause(x,0,y,0);	
			}
			else add_clause(x,1,y,1);
		}
		if(!jud())
		{
			cout<<"No solution"<<endl;
			continue;
		}
		for(int i = 0;i < n;i++)
		 if(num[i] == 1)
		 {
		 	if(mark[2*i]) cout<<"A"<<endl;
		 	else cout<<"C"<<endl;
		 }
		 else 
		  if(mark[2*i]) cout<<"B"<<endl;
		  else cout<<"C"<<endl;
	}
}