题意:平面上l盏灯,每盏灯可以照亮横向的2*r+1个格子或者纵向的2*r+1个格子,让你确定每盏灯的方向,使得每个格子只被同一行的不超过一盏灯照亮,并且只被同一列的不超过一盏灯照亮。输出是否有解。
显然的2-sat模型。
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
vector<int>G[2005],rG[2005],vs;
int a[1005][1005];
int n,r,l,xs[1005],ys[1005];
void AddEdge(int U,int V){
G[U].push_back(V);
rG[V].push_back(U);
}
bool used[2005];
void dfs(int U){
used[U]=1;
for(int i=0;i<G[U].size();++i){
if(!used[G[U][i]]){
dfs(G[U][i]);
}
}
vs.push_back(U);
}
int K,cmp[2005];
void rdfs(int U){
used[U]=1;
cmp[U]=K;
for(int i=0;i<rG[U].size();++i){
if(!used[rG[U][i]]){
rdfs(rG[U][i]);
}
}
}
int main(){
// freopen("f.in","r",stdin);
scanf("%d%d%d",&n,&r,&l);
for(int i=1;i<=l;++i){
scanf("%d%d",&xs[i],&ys[i]);
a[xs[i]][ys[i]]=i;
}
for(int i=1;i<=l;++i){
for(int j=1;j<=2*r && xs[i]+j<=n;++j){
if(a[xs[i]+j][ys[i]]){
AddEdge(i,a[xs[i]+j][ys[i]]+l);
}
}
for(int j=1;j<=2*r && xs[i]-j>=1;++j){
if(a[xs[i]-j][ys[i]]){
AddEdge(i,a[xs[i]-j][ys[i]]+l);
}
}
for(int j=1;j<=2*r && ys[i]+j<=n;++j){
if(a[xs[i]][ys[i]+j]){
AddEdge(i+l,a[xs[i]][ys[i]+j]);
}
}
for(int j=1;j<=2*r && ys[i]-j>=1;++j){
if(a[xs[i]][ys[i]-j]){
AddEdge(i+l,a[xs[i]][ys[i]-j]);
}
}
}
for(int i=1;i<=2*l;++i){
if(!used[i]){
dfs(i);
}
}
memset(used,0,sizeof(used));
for(int i=vs.size()-1;i>=0;--i){
if(!used[vs[i]]){
++K;
rdfs(vs[i]);
}
}
for(int i=1;i<=l;++i){
if(cmp[i]==cmp[i+l]){
puts("NO");
return 0;
}
}
puts("YES");
return 0;
}