【UVALive

题意：

　　有n个宇航员，按照年龄划分，年龄低于平均年龄的是年轻宇航员，而年龄大于等于平均年龄的是老练的宇航员。

　　现在要分配他们去A，B，C三个空间站，其中A站只有老练的宇航员才能去，而B站是只有年轻的才能去，C站都可以去。

　　有m对宇航员相互讨厌，不能让他们在同一个空间站工作。

　　输出每个宇航员应分配到哪个空间站，如果没有则输出No solution.

分析：

　　对于每个宇航员，有两种选择，（A,B）或C。第一个选择中取A还是取B取决于年龄。

　　构图，2-SAT找满足题意的方案再输出即可。

代码如下：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#define Maxn 100010
#define Maxm 100010

int n,m;
int age[Maxn],first[2*Maxn],mark[2*Maxn],s[2*Maxn];
int c,v;

struct node
{
    int x,y,next;
}t[4*Maxm];int len;

void ins(int x,int y)
{
    t[++len].x=x;t[len].y=y;
    t[len].next=first[x];first[x]=len;
}

bool dfs(int x)
{
    if(mark[x^1]) return 0;
    if(mark[x]) return 1;
    mark[x]=1;
    s[++c]=x;
    for(int i=first[x];i;i=t[i].next)
     if(!dfs(t[i].y)) return 0;
    return 1;
}

bool solve()
{
    memset(mark,0,sizeof(mark));
    for(int i=0;i<n;i++)
     if(!mark[2*i]&&!mark[2*i+1])
     {
         c=0;
         if(!dfs(2*i))
         {
             while(c>0) mark[s[c--]]=0;
             if(!dfs(2*i+1)) return 0;
        }
      }
      return 1;
}

void output()
{
    for(int i=0;i<n;i++)
     if(mark[i*2+1]) printf("C
");
     else if(age[i]*n>=v) printf("A
");
     else printf("B
");
}

int main()
{
    while(1)
    {
        
        scanf("%d%d",&n,&m);
        len=0;v=0;
        if(n==0&&m==0) break;
        memset(first,0,sizeof(first));
        for(int i=0;i<n;i++) scanf("%d",&age[i]),v+=age[i];
        for(int i=1;i<=m;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);x--;y--;
            ins(x*2+1,y*2);ins(y*2+1,x*2);
            if(!((age[x]*n>=v)^(age[y]*n>=v))) ins(x*2,y*2+1),ins(y*2,x*2+1);
        }
        if(!solve()) printf("No solution.
");
        else output();
    }
    return 0;
}

2016-03-18 13:19:17