UVALive - 3713 Astronauts

给定n个宇航员的年龄,平均年龄为 ave,根据下列要求分配任务:

B任务只能分配给年龄<ave的宇航员;

A任务只能分配给年龄>=ave的宇航员;

C任务可以任意分配。

给定m组互相憎恨的宇航员,要求他们不能分配到同一个任务。能否存在这样的一组任务分配。

每个宇航员都只能分配两种任务中的一种:A或C(年龄大于等于ave),B或C(年龄小于ave),那么为每个宇航员设立一个变量xi,xi为0表示分配C任务,为1则分配A或B(根据年龄)。

对于互相仇恨的宇航员,如果属于同一类型,那么应满足xi∨xj,非xi∨非xj,表示xi和xj一真一假;如果类型不同只需要满足不同时分配C任务就可:xi∨xj。

 

直接贴代码好了:

#include <iostream>
#include <sstream>
#include <cstdio>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <string>
#include <stack>
#include <map>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
#define esp 1e-6
#define pi acos(-1.0)
#define pb push_back
#define mp(a, b) make_pair((a), (b))
#define in  freopen("in.txt", "r", stdin);
#define out freopen("out.txt", "w", stdout);
#define print(a) printf("%d\n",(a));
#define bug puts("********))))))");
#define stop  system("pause");
#define Rep(i, c) for(__typeof(c.end()) i = c.begin(); i != c.end(); i++)
#define pragma comment(linker, "/STACK:102400000, 102400000")
#define inf 0x0f0f0f0f

using namespace std;
typedef long long  LL;
typedef vector<int> VI;
typedef pair<int, int> pii;
typedef vector<pii,int> VII;
typedef vector<int>:: iterator IT;
const int maxn = 111111;
int A[maxn], tag[maxn];
double sum;
int dblcmp(int x)
{
    if(fabs(x) < esp)
        return 1;
    return x > 0 ? 1 : 0;
}

struct TwoSAT
{
    int mark[maxn*2], S[maxn*2], c, n;
    VI g[maxn*2];

    void init(int n)
    {
        this->n = n;
        for(int i = 0; i < n*2; i++)
            g[i].clear();
        memset(mark, 0, sizeof(mark));
    }
    bool dfs(int x)
    {
        if(mark[x^1]) return false;
        if(mark[x])    return true;
        mark[x] = 1;
        S[c++] = x;
        for(int i = 0; i < g[x].size(); i++)
            if(!dfs(g[x][i])) return false;
        return true;
    }
    void add_clause(int x, int xval, int y, int yval)
    {
        x = x*2+xval;
        y = y*2+yval;
        g[x^1].pb(y);
        g[y^1].pb(x);
    }
    bool solve()
    {
        for(int i = 0; i < 2*n; i += 2)
            if(!mark[i] && !mark[i+1])
            {
                c = 0;
                if(!dfs(i))
                {
                    while(c) mark[S[--c]] = false;
                    if(!dfs(i+1)) return false;
                }
            }
        return true;
    }
    void print_ans(void)
    {
        for(int i = 0; i < n; i++)
        {
            puts(mark[2*i] ? "C" : (tag[i] == 1 ? "A" : "B"));
        }
    }
} sat;
bool check(int x)
{
    return dblcmp(A[x]-sum);
}
int main(void)
{   
    int n, m;
    while(scanf("%d%d", &n, &m), n||m)
    {
        sum = 0;
        for(int i = 0; i < n; i++)
            scanf("%d", A+i), sum += A[i];
        sum /= n;
        sat.init(n);
        for(int i = 0; i < n; i++)
            tag[i] = check(i);
        while(m--)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            u--, v--;
            if(tag[u] != tag[v])
            {
                sat.add_clause(u, 1, v, 1);
            }
            else
            {
                sat.add_clause(u, 1, v, 1);
                sat.add_clause(u, 0, v, 0);
            }
        }
        if(sat.solve())
            sat.print_ans();
        else puts("No solution.");
    }
    return 0;
}

你可能感兴趣的:(live)