POJ-2749-Building roads(2-sat)

POJ 2-sat六题最后一题
http://blog.sina.com.cn/s/blog_64675f540100k2xf.html
poj 六道2-sat最后一题第六题：

题目描述：有n个农场，每个农场有坐标x，y。

有两个集合点s1和s2（也有坐标），每个农场必须连接其中的一个（有且仅有一个）。

然后有A个条件，每个条件a，b表示a农场不能和b农场连接在一个集合点。

然后再有B个条件，每个条件a，b表示a农场必须和b农场连接在一个集合点。

问你，在各种合法的连接情况中，任何两个农场间的距离的最大值的最小值是多少。

解题报告：

每个农场i分成两个点，i和i + n，前面表示连接左侧集合点，后面的表示连接右侧集合点。

对于条件A中的ab，连接

<a, b + n> <a + n, b> <b, a + n> <b + n, a>

这样就保证了ab不再一个集合点。同理，B条件也很好写。

然后就是用二分枚举答案（距离的最大值）。

对于每一次枚举的答案key，枚举任意两个农场a,b, 如果他俩的4种距离（a->s1->s1>b, a->s2->s2->b, a->s1->s2->b, a->s2->s1->b)有大于key的，就加入判定条件，不能这样连接。

// File Name: 2749.cpp
// Author: zlbing
// Created Time: 2013/2/5 21:18:58

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<stack>
#include <cmath>
using namespace std;
#define MAXN 1000
struct TwoSAT{
int n;
vector<int>G[MAXN*2];
bool mark[MAXN*2];
stack<int>S;
bool dfs(int x)
{
if(mark[x^1])return false;
if(mark[x])return true;
mark[x]=true;
S.push(x);
for(int i=0;i<G[x].size();i++)
{
int v=G[x][i];
if(!dfs(v))return false;
}
return true;
}
void init(int _n)
{
n=_n;
for(int i=0;i<2*n;i++)
G[i].clear();
memset(mark,0,sizeof(mark));
}
void add_clause(int x,int y)
{
G[x].push_back(y);
}
bool solve()
{
for(int i=0;i<2*n;i=i+2)
{
if(!mark[i]&&!mark[i+1]){
while(!S.empty())
{
S.pop();
}
if(!dfs(i))
{
while(!S.empty())
{
mark[S.top()]=false;
S.pop();
}
if(!dfs(i+1))return false;
}
}
}

//        for(int i=0;i<2*n;i++)
//            if(mark[i])printf("%d ",T[i/2][i%2]);
//        printf("\n");
return true;
}
};
int N,A,B;
int s[2][2];
int d[MAXN][2];
int d1[MAXN][2];
int d2[MAXN][2];
int AlenB;
TwoSAT solver;
int count(int i,int j)
{
return abs(d[i][0]-s[j][0])+abs(d[i][1]-s[j][1]);
}
bool test(int M)
{
solver.init(N);
for(int i=0;i<A;i++)
{
solver.add_clause(d1[i][0]*2,d1[i][1]*2+1);
solver.add_clause(d1[i][0]*2+1,d1[i][1]*2);
solver.add_clause(d1[i][1]*2+1,d1[i][0]*2);
solver.add_clause(d1[i][1]*2,d1[i][0]*2+1);
}
for(int i=0;i<B;i++)
{
solver.add_clause(d2[i][0]*2,d2[i][1]*2);
solver.add_clause(d2[i][0]*2+1,d2[i][1]*2+1);
solver.add_clause(d2[i][1]*2,d2[i][0]*2);
solver.add_clause(d2[i][1]*2+1,d2[i][0]*2+1);
}
for(int i=0;i<N;i++)
for(int j=i+1;j<N;j++)
{
if(count(i,0)+count(j,0)>M)
{
solver.add_clause(i*2,j*2+1);
solver.add_clause(j*2,i*2+1);
}
if(count(i,1)+count(j,1)>M)
{
solver.add_clause(i*2+1,j*2);
solver.add_clause(j*2+1,i*2);
}
if(count(i,1)+count(j,0)+AlenB>M)
{
solver.add_clause(i*2+1,j*2+1);
solver.add_clause(j*2,i*2);
}
if(count(i,0)+count(j,1)+AlenB>M)
{
solver.add_clause(i*2,j*2);
solver.add_clause(j*2+1,i*2+1);
}
}
if(solver.solve())return true;
else return false;
}
#define MAXN 4000000
int main(){
while(~scanf("%d%d%d",&N,&A,&B))
{
scanf("%d%d%d%d",&s[0][0],&s[0][1],&s[1][0],&s[1][1]);
AlenB=abs(s[0][0]-s[1][0])+abs(s[0][1]-s[1][1]);
for(int i=0;i<N;i++)
scanf("%d%d",&d[i][0],&d[i][1]);
int a,b;
for(int i=0;i<A;i++)
{
scanf("%d%d",&a,&b);
d1[i][0]=--a;
d1[i][1]=--b;
}
for(int i=0;i<B;i++)
{
scanf("%d%d",&a,&b);
d2[i][0]=--a;
d2[i][1]=--b;
}
int L=0,R=MAXN;
while(L<R)
{
int mid=L+(R-L)/2;
if(test(mid)){
R=mid;
}
else L=mid+1;
}
if(R==MAXN)printf("-1\n");
else
printf("%d\n",R);
}
return 0;
}