Poj 3905 Perfect Election（2-sat 可行性判定）

- 日理万妓 2022-08-11 00:40 127阅读 0赞

基础2-sat题目

这题有一个收获就是意识到带正负号的整数可以用 %d 直接读入，不需要字符串！

#include <cstdio>
    #include <cstring>
    #include <algorithm>
    using namespace std;
    
    #define max(a,b) ((a)>(b)?(a):(b))
    #define min(a,b) ((a)<(b)?(a):(b))
    
    const int MAXN = 1010;  
    const int VN   = MAXN*2;  
    const int EN   = VN*VN*2; 
    int n,m;
    
    struct Edge{
    	int v, next;
    };
    
    class Graph{
    public:
    	int head[VN],size;
    	Edge E[EN];
    	void init(){
    		size = 0;
    		memset(head, -1, sizeof(head));
    	}
    	void Add (int u,int v){
    		E[size].v = v;
    		E[size].next = head[u];
    		head[u] = size++;
    	}
    }g;
    
    class Two_Sat{
    public:
    	bool Check(const Graph&g, const int n){
    		SCC(g,n);
    		for (int i=0;i<n;i++)
    			if (belong[i*2] == belong[i*2+1])
    				return false;
    		return true;
    	}
    private:
    	int top, bcnt, idx;
    	int sta[VN],belong[VN];
    	int dfn[VN],low[VN];
    	bool inStack[VN];
    	
    	void targan (const Graph&g, const int u){
    		int v;
    		dfn[u] = low[u] = ++idx;
    		sta[top++] = u;
    		inStack[u] = true;
    		
    		for (int i=g.head[u]; i!=-1; i=g.E[i].next){
    			v = g.E[i].v;
    			if (dfn[v] < 0){
    				targan(g, v);
    				low[u] = min(low[u], low[v]);
    			}
    			else if(inStack[v])
    				low[u] = min(low[u], dfn[v]);
    		}
    		
    		if (dfn[u] == low[u]){
    			++bcnt;
    			do{
    				v = sta[--top];
    				inStack[v] = false;
    				belong[v] = bcnt;
    			}while(u != v);
    		}
    	}
    
    	void SCC (const Graph&g, int n){
    		top=bcnt=idx=0;
    		memset(dfn,-1,sizeof(dfn));
    		memset(low,0,sizeof(low));
    		memset(belong,0,sizeof(belong));
    		memset(inStack,false,sizeof(inStack));
    		for (int i=0; i<2*n;i++)
    			if (dfn[i] < 0) targan(g,i);
    	}
    }sat;
    
    int main ()
    {
    	while (~scanf("%d%d", &n, &m))
    	{
    		g.init();
    		int a,b,u,v;
    		for (int i=0;i<m;i++)
    		{
    			scanf("%d%d",&a,&b);
    			
                if (a>0 && b>0)
    			{ // + + 至少取一个
    				u=(a-1)*2,v=(b-1)*2;
    				g.Add(u^1,v);
    				g.Add(v^1,u);
    			}
    			else if (a<0 && b<0)
    			{// - -  至少有一个不取
    				a=-a,b=-b;
    				u=(a-1)*2,v=(b-1)*2;
    				g.Add(u,v^1);
    				g.Add(v,u^1);
    			} 
    			else if (a>0 && b<0)
    			{// + -  选A或不选B或同时发生,也就是!A和B不能同时选
    				b=-b;
    				u=(a-1)*2,v=(b-1)*2;
    				g.Add(u^1,v^1);
    				g.Add(v,u);
    			}
    			else if (a<0 && b>0)
    			{ // - +  
    				a=-a;
    				u=(a-1)*2,v=(b-1)*2;
    				g.Add(u,v);
    				g.Add(v^1,u^1);  
    			}
    		}
    		if(sat.Check(g, n)) puts("1");  
    		else puts("0");  
    	}
    	return 0;  
    }