UVALive 3713 浅谈2-SAT问题图论求解法

世界真的很大
2-SAT问题是一种神奇的问题，看似和图论不太有关的问题却可以用图论的算法解决
其实大概是题目要求中有关乎于条件与条件之间的关系，就可以想到建边然后跑一下奇奇怪怪的图论算法
2-SAT大概是一种适定性问题，具体什么也不太懂，而且3-SAT及其以上的4-SAT，5-SAT都已经被证明是一种NP完全问题了
看题先：
description：

大概是有很多个宇航员，有三种任务A，B，C，小于宇航员的平均年龄的宇航员才能选择B任务，大于等于的才能选择A任务，而C任务没有限制。
而某一些宇航员之间互相讨厌，所以不能选择同一种任务，问一种可行的方案

input

多组数据，到0 0为止
每组数据第一行包含2个整数n，m，表示一共有n个宇航员，m对宇航员互相讨厌

output

每组数据包含n行，第i行一个字母表示第i名宇航员选择的任务

简单来看，虽然一共有3种任务，但是对于每个宇航员来讲，只有两种选择，且不同宇航员选择之间会有冲突
这就是一种典型的2-SAT问题了，话说今年的NOI DAY2 T1好像就是2-SAT？
先看看这两篇博客，讲的比较详细：
第一篇，第二篇
简单来说，把A，B任务归为一类的话，每个人有两种选择，选C，不选C（选A，B），假设有两个人X，Y互相讨厌，如果X选C的话，Y就一定不能选C，即X选C，Y一定选（A，B）。
甚者，如果X，Y在一个年龄段的话，那么X选（A，B），Y必须选C
这便是宇航员之间的“关系”。
考虑将这种关系具体化，成为“边”，对于每个宇航员，映射两个点，一个表示这个宇航员选C，另一个表示其选（A，B），一个点向另一个点连边，表示如果这个点被选了，那另一个点也必须被选，比如就是X选C的点要向Y的（A，B）连边。
这样就能得到一张像模像样的图了。现在在考虑方案之前，首先考虑合法性
对于一个点X，他对应的两个点能且只能选择一个，因为要么（A，B），要么C对吧
如果选择（A，B），一路推下去，使得C也必须被选，并且如果不选（A，B），一路推下去，C也不选，那么这张图就不可能有合法方案了。这就是说，在（A，B）与C之间存在要么都选，要么都不选的情况，就不合法了，而这样一来，因为是有向图，（A，B）与C一定在一个强连通分量里了。
对图跑一边TARJAN，缩点之后先判断每个宇航员1对应的两个点在不在同样的强连通分量里面，如果在，输出“No solution.”就行了。
那么处理有可行解的情况，缩点之后原图就是一张DAG了，可以对其建立拓扑序，这样就得到了如果选了某一个点，哪些点是不受影响的，就是那些拓扑序比他小的点，哪些点会受到影响，只可能是在其拓扑序之后的点了
之前的那篇博客里说干脆建一个反图，其实不用的
在按拓扑序访问时，对于每一个宇航员，他的两个点如果都没有访问，而这时访问到了其中一个点，证明他的拓扑序一定比另一个靠前，为了使其影响到的方案尽量少，我们优先选择后面的点，所以把先访问到的点标记为不选，而把另一个标记为选择
注意缩点完了后是需要重新建边的
而且这道题是任意一种方案都可以的，所以输出和样例输出不一样就不要在意了2333
完整代码：

#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<stack>
#include<queue>
using namespace std;

struct edge
{
    int u,v,last;
}ed[1000010],ad[1000010];

queue <int> state;
stack <int> stk;

int head[200010],mrk[200010],dfn[200010],in[200010],age[200010];
int low[2000010],ins[200010],place[200010],rev[200010];
int n,m,num,tot,cnt,idx,mum,flag;

int init()
{
    num=0,tot=0,cnt=0,idx=0,mum=0,flag=0;
    while(!stk.empty()) stk.pop();
    memset(in,0,sizeof(in));
    memset(age,0,sizeof(age));
    memset(head,0,sizeof(head));
    memset(low,0,sizeof(low));
    memset(mrk,-1,sizeof(mrk));
    memset(dfn,0,sizeof(dfn));
    memset(ins,0,sizeof(ins));
    memset(place,0,sizeof(place));
    memset(rev,0,sizeof(rev));
}

void add(int u,int v)
{
    num++;
    ed[num].v=v;
    ed[num].u=u;
    ed[num].last=head[u];
    head[u]=num;
}

void ade(int u,int v)
{
    mum++;
    ad[mum].v=v;
    ad[mum].last=head[u];
    head[u]=mum;
    in[v]++;
}

void tarjan(int u)
{
    dfn[u]=low[u]=++idx;
    ins[u]=1;
    stk.push(u);
    for(int i=head[u];i;i=ed[i].last)
    {
        int v=ed[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(ins[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(dfn[u]==low[u])
    {
        cnt++;
        int t=-1;
        while(t!=u)
        {
            t=stk.top();
            ins[t]=0,place[t]=cnt;
            stk.pop();
        }
    }
}

void top()
{
    while(!state.empty()) state.pop();
    for(int i=1;i<=cnt;i++)
        if(!in[i])      
            state.push(i);          
    while(!state.empty())
    {
        int u=state.front();
        state.pop();
        if(mrk[u]==-1) mrk[u]=1,mrk[rev[u]]=2;
        for(int i=head[u];i;i=ad[i].last)
        {
            int v=ad[i].v;
            in[v]--;
            if(in[v]==0)
                state.push(v);  
        }
    }
}

void rebuild()
{
    memset(head,0,sizeof(head));
    for(int i=1;i<=num;i++)
    if(place[ed[i].u]!=place[ed[i].v])
        ade(place[ed[i].v],place[ed[i].u]);
    for(int i=1;i<=n;i++)
        rev[place[i*2]]=place[i*2+1],rev[place[i*2+1]]=place[i*2];
}

int belong(int x)
{
    return n*age[x] < tot;
}

int main()
{
    while(1)
    {
        init();
        scanf("%d%d",&n,&m);
        if(!n) break ;
        for(int i=1;i<=n;i++)
            scanf("%d",&age[i]),tot+=age[i];
        for(int i=1;i<=m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(a==b) continue ;         
            add(2*a,2*b+1);
            add(2*b,2*a+1);
            if( belong(a) == belong(b) )
            {
                add(2*a+1,2*b);
                add(2*b+1,2*a);
            }
        }
        for(int i=2;i<=2*n+1;i++)
            if(!dfn[i]) tarjan(i);
        for(int i=1;i<=n;i++)
            if(place[2*i]==place[2*i+1])
            {
                puts("No solution.");
                flag=1;
                break ;
            }
        if(flag) continue ;
        rebuild();
        top();
        for(int i=1;i<=n;i++)
        {
            if(mrk[place[2*i]]==1) puts("C");
            else if( belong(i) ) puts("B");
            else puts("A");
        }
    }
    return 0;
}
/*
I am the bone of my Code
Unknown to death, nor unknown to life
As I pray, Unlimited AC Codes.
*/

嗯，就是这样