hdu1824 Let's go home（2-sat）

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 
 4 const int N=4e3+5,M=5e3+5;
 5 int low[N],dfn[N],Stack[N],belong[N];
 6 bool inStack[N];
 7 int n,m,tot,tag,top;
 8 vector<int> G[N];
 9 map<int,int> mp;
10 
11 void init(){
12     top=tag=0;
13     memset(low,0,sizeof low);
14     memset(dfn,0,sizeof dfn);
15     memset(Stack,0,sizeof Stack);
16     memset(belong,0,sizeof belong);
17     memset(inStack,false,sizeof inStack);
18     for(int i=0;i<N;i++){
19         G[i].clear();
20     }
21     mp.clear();
22 }
23 
24 void tarjan(int u){
25     int v;
26     low[u]=dfn[u]=++tot;
27     Stack[++top]=u;
28     inStack[u]=true;
29     for(int i=0;i<G[u].size();i++){
30         v=G[u][i];
31         if(dfn[v]==0){
32             tarjan(v);
33             low[u]=min(low[u],low[v]);
34         }
35         else if(inStack[v]==true){
36             low[u]=min(low[u],dfn[v]);
37         }
38     }
39     if(dfn[u]==low[u]){
40         tag++;
41         do{
42             v=Stack[top--];
43             inStack[v]=false;
44             belong[v]=tag;
45         }while(u!=v);
46     }
47 }
48 
49 int main(){
50     int x,y,z;
51     while(~scanf("%d%d",&n,&m)){
52         init();
53         for(int i=0,k=0;i<n;i++){
54             scanf("%d%d%d",&x,&y,&z);
55             mp[x]=k++,mp[y]=k,mp[z]=k++;
56             G[mp[x]*2].push_back(mp[y]*2+1);
57             G[mp[y]*2+1].push_back(mp[x]*2);
58             G[mp[y]*2].push_back(mp[x]*2+1);
59             G[mp[x]*2+1].push_back(mp[y]*2);
60         }
61         for(int i=0;i<m;i++){
62             scanf("%d%d",&x,&y);
63             G[mp[x]*2].push_back(mp[y]*2+1);
64 //            G[mp[y]*2+1].push_back(mp[x]*2);
65             G[mp[y]*2].push_back(mp[x]*2+1);
66 //            G[mp[x]*2+1].push_back(mp[y]*2);
67         }
68         for(int i=0;i<4*n;i++){
69             if(dfn[i]==0){
70                 tot=0;
71                 tarjan(i);
72             }
73         }
74         int flag=1;
75         for(int i=0;i<4*n;i+=4){
76             if(belong[i]==belong[i+2]){
77                 flag=0;
78                 break;
79             }
80         }
81         if(flag==1)printf("yes\n");
82         else printf("no\n");
83     }
84 }