UVA Live 3713 Astronauts （2-SAT）

用布尔变量表示状态，把限制条件转化为XνY的形式以后跑2SAT，根据变量取值输出方案。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+;

#define PB push_back
bool vis[maxn*];
vector<int> G[maxn*];
int N,S[maxn*],top;
void initGraph(int n)
{
    N = n*;
    for(int i = ; i < N; i++) G[i].clear();
    memset(vis,,sizeof(bool)*(n<<|));
}

bool dfs(int x)
{
    if(vis[x^]) return false;
    if(vis[x]) return true;
    vis[x] = true;
    S[top++] = x;
    for(int i = ; i < (int)G[x].size(); i++){
        if(!dfs(G[x][i])) return false;
    }
    return true;
}

bool twoSAT()
{
    for(int i = ; i < N; i+=){
        if(!vis[i]&&!vis[i^]){
            top = ;
            if(!dfs(i)){
                while(top>) vis[S[--top]] = false;
                if(!dfs(i^)) return false;
            }
        }
    }
    return true;
}

void add_clause(int x,int xv,int y,int yv)
{
    x = x<<|xv;
    y = y<<|yv;
    G[x^].PB(y);
    G[y^].PB(x);
}

int tp[maxn];
int age[maxn];
char bin[][] = {{'C','B'},{'C','A'}};

int main()
{
    //freopen("in.txt","r",stdin);
    int n,m;
    while(scanf("%d%d",&n,&m),n){
        int sum = ;
        for(int i = ; i < n; i++){
            scanf("%d",age+i);
            sum += age[i];
        }
        for(int i = ; i < n; i++){
            if(age[i]*n >= sum) tp[i] = ;
            else tp[i] = ;
        }
        initGraph(n);
        for(int i = ; i < m; i++){
            int u,v; scanf("%d%d",&u,&v);
            if(tp[--u]^tp[--v]){
                add_clause(u,,v,);
            }else {
                add_clause(u,,v,);
                add_clause(u,,v,);
            }
        }
        if(twoSAT()){
            for(int i = ; i < N; i+=){
                putchar(bin[tp[i>>]][vis[i^]]);
                putchar('\n');
            }
        }else {
            puts("No solution.");
        }
    }
    return ;
}