P4782 【模板】2-SAT 问题

题目链接

2-sat问题，看这个题解：（超级详细！！！）
链接

注意这里的scc[]数组，和id[] dfn[] low[]数组是不完全同的！
注意点数要开到2n，边数要开到4n

代码

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<map>
#include<algorithm>
#include<deque>
#include<stack>
#include<set>
// #include <unordered_map>
#include<math.h>
#include<string.h>
#define IOS ios::sync_with_stdio(false),cin.tie(0);
using namespace std;
 
#define pb push_back
#define coutl cout<<"------------"<<endl;
#define fi first
#define se second

#define ire(x) scanf("%d",&x)
#define iire(a,b) scanf("%d %d",&a,&b)
#define lre(x) scanf("%lld",&x)
#define llre(a,b) scanf("%lld %lld",&a,&b)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define endl "\n"
#define PI acos(-1.0)
//#define int long long
typedef long long ll;
typedef unsigned long long ull;
      
typedef pair<int, int> PII;
typedef pair<double, int> PDI;
typedef pair<ll, ll> PLL;
typedef pair<double, double> PDD;
typedef pair<double, pair<int, double> > PDID;
typedef pair<char, char> PCC;
typedef pair<char, pair<int, int> > PCII;
typedef pair<int, pair<int, int> > PIII;
typedef pair<int, pair<int, pair<int, int> > > PIIII;
typedef pair<ll, pair<int, int> > PLII;

const int maxn = 2e5 + 7;
const int N = 2e6 + 7;
const int M = 4e6 + 7;
const int mod = 3*5*7*11*13*17*19*23;
const int inv = mod - mod/2;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1);
const double eps = 1e-8;
  
ll gcd(ll a,ll b) {return b==0 ? a : gcd(b,a%b);}
ll lcm(ll a,ll b) {return a*b / gcd(a,b);}
ll qmi(ll a,ll b,ll p) {ll ans = 1; while(b) { if(b & 1) ans = ans * a % p; a = a * a % p; b >>= 1; } return ans;}
int lowbit(int x) {return x & (-x);}

int n,m;
int h[N],e[M],ne[M],idx;    //原始图

//定义两个时间戳（当dfn[u] == low[u]是，就表示u是其所在的强连通分量的最高点）
int dfn[N];     //dfn[u]表示遍历到u的时间戳
int low[N];     //low[u]表示从u开始走，所能遍历到的最小时间戳
int timestamp;  //时间戳

//栈中元素理解为: 该点被遍历过，但是其所在的强连通分量没被遍历完
stack<int> st; 
bool in_stack[N];   //标记一下每个点是否在栈中

int scc_cnt;    //强连通分量编号
int id[N];      //记录一下每个点属于哪个强连通分量
int sizes[N];    //每个强连通分量里面有多少个点
int dout[N];    //记录每个强连通分量的出度是多少
int scc[N];

void add(int a,int b)
{
	e[idx]=b; ne[idx]=h[a]; h[a]=idx++;
}

void targan(int u)
{
    dfn[u] = low[u] = ++timestamp;  //每次先求一下每个点的时间戳是多少
    st.push(u);     //进栈
    in_stack[u] = true;    //标记 

    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j = e[i];
        if(!dfn[j]) //如果没被遍历过
        {
            targan(j);  //遍历
            low[u] = min(low[u],low[j]);    //回溯的时候更新u的low[]
        }
        else if(in_stack[j]) //还在栈中，表示其强连通分量还没遍历完，则继续更新
            low[u] = min(low[u],dfn[j]);
    }

    //说明当前这个点是强连通分量的最高点(该强连通分量遍历完)
    if(low[u] == dfn[u])    
    {
        int y;
        ++ scc_cnt;

        do
        {
            y = st.top();
            st.pop();   //该点出栈
            in_stack[y] = false;
			
			scc[y] = scc_cnt;
            id[y] = scc_cnt;    //标记一下该点的强连通分量编号
            sizes[scc_cnt] ++;   //强连通分量的点数++

        }while(y != u);     //y=u时说明当前强连通分量处理完了
    }
}

int main()
{
    memset(h, -1, sizeof h);
    
	iire(n,m);
	while(m--)
	{
		int i,a,j,b;	//i表示false，i+n表示true
		iire(i,a);
		iire(j,b);
		
		if(a && b)	//(i=1 || j=1) => (i=0 -> j=1) && (j=0 -> i=1)
		{
			add(i,j+n);
			add(j,i+n);
		}
		else if(!a && b)	//(i=0 || j=1) => (i=1 -> j=1) && (j=0 -> i=0)
		{
			add(i+n,j+n);
			add(j,i);
		}
		else if(a && !b)	//(i=1 || j=0) => (i=0 -> j=0) && (j=1 -> i=1)
		{
			add(i,j);
			add(j+n,i+n);
		}
		else if(!a && !b)	//(i=0 || j=0) => (i=1 -> j=0) && (j=1 -> i=0)
		{
			add(i+n,j);
			add(j+n,i);
		}
	}
	
	//对每个变量tarjan一下
	for(int i=1;i<=2*n;i++)
		if(!dfn[i]) targan(i);
	
	//判断无解（一个变量的两种取值在同一连通分量中）
	for(int i=1;i<=n;i++)
		if(scc[i] == scc[i+n])
		{
			printf("IMPOSSIBLE\n");
			return 0;
		}
	
	//有解，输出拓扑序大的取值，注意这里tarjan完的拓扑序是反向的
	printf("POSSIBLE\n"); 
	for(int i=1;i<=n;i++)	//
		if(scc[i+n] < scc[i]) cout<<"1 ";
		else cout<<"0 ";
	
	return 0;
}