UVA1391【LA3713】 经典2-SAT

题意：给出一些宇航员他们的年龄，x是他们的平均年龄，其中A任务只能给年龄大于等于x的人，B任务只能给小于x的人，C任务没有限制。再给出m对人，他们不能同任务。现在要你输出一组符合要求的任务安排。

这道题是刘汝佳蓝书上讲了的题，在这里套用他的模板，实在是好用啊…

#include<stdio.h>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=100005;
double sum,ave;
int age[maxn],u,v,n,m;
inline int read(){
    register int x=0,f=1;
    register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f*=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return f*x;
}
struct SAT{
    int n;
    vector<int> G[maxn*2];
    bool mark[maxn*2];
    int s[maxn*2],c;
    bool dfs(int x){
        if(mark[x^1]) return false;
        if(mark[x]) return true;
        mark[x]=true;
        s[c++]=x;
        for(int i=0;i<G[x].size();i++)
         if(!dfs(G[x][i])) return false;
        return true;
    }
    void init(int n){
        this->n=n;
        for(int i=0;i<n*2;i++) G[i].clear();
        memset(mark,false,sizeof(mark));
    }
    void add(int x,int xval,int y,int yval){
        x=x*2+xval;
        y=y*2+yval;
        G[x^1].push_back(y);
        G[y^1].push_back(x);
    }
    bool solve(){
        for(int i=0;i<n*2;i=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;
    }

};
SAT Mercer;
int main(){
    while(n=read(),m=read(),n||m){
       Mercer.init(n);sum=0;
       for(register int i=0;i<n;i++) age[i]=read(),sum+=age[i];
       ave=sum/n;
       for(register int i=0;i<m;i++){
           u=read(),v=read();
           u--,v--;
           if((age[u]<ave&&age[v]<ave)||(age[u]>=ave&&age[v]>=ave)){
              Mercer.add(u,0,v,0);
              Mercer.add(u,1,v,1);
           }
           else Mercer.add(u,1,v,1);
       }
       if(!Mercer.solve()) puts("No solution.");
       else{
          for(int i=0;i<2*n;i=i+2){
              if(Mercer.mark[i])
               puts("C");
               else printf("%c\n",age[i/2]>=ave?'A':'B');
          }
       }
    }
    return 0;
}