POJ 3740 Easy Finding (DLX模板)
Description
Given a
M×
N matrix
A.
A
ij ∈ {0, 1} (0 ≤ i < M, 0 ≤ j < N), could you find some rows that let every cloumn contains and only contains one 1.
Input
There are multiple cases ended by
EOF. Test case up to 500.The first line of input is
M,
N (
M ≤ 16,
N ≤ 300). The next
M lines every line contains
N integers separated by space.
Output
For each test case, if you could find it output "Yes, I found it", otherwise output "It is impossible" per line.
Sample Input
3 3 0 1 0 0 0 1 1 0 0 4 4 0 0 0 1 1 0 0 0 1 1 0 1 0 1 0 0
Sample Output
Yes, I found it It is impossible
直接套用DLX模板,C++所用时间刚好差不多
#include <iostream>
#include <string.h>
using namespace std;
struct DLX
{
int n,m,size;
int U[5000],D[5000],R[5000],L[5000],Row[5000],Col[5000];
int H[18];//行头节点
int S[310];//每列有多少个节点
int ansd,ans[18];//如果有答案，则选了ansd行，具体是哪几行放在ans[ ]数组里面，ans[0~ansd-1];
void init(int _n,int _m)
{
n=_n,m=_m;
for(int i=0; i<=m; i++)
{
S[i]=0;
U[i]=D[i]=i;//初始状态下，上下自己指向自己
L[i]=i-1;
R[i]=i+1;
}
R[m]=0,L[0]=m;
size=m;//编号，每列都有一个头节点，编号1-m
for(int i=1; i<=n; i++)
H[i]=-1;//每一行的头节点
}
void link(int r,int c)//第r行，第c列
{
++S[Col[++size]=c];//第size个节点所在的列为c,当前列的节点数++
Row[size]=r;//第size个节点行位置为r
D[size]=D[c];//下面这四句头插法（图是倒着的？)
U[D[c]]=size;
U[size]=c;
D[c]=size;
if(H[r]<0)
H[r]=L[size]=R[size]=size;
else
{
R[size]=R[H[r]];
L[R[H[r]]]=size;
L[size]=H[r];
R[H[r]]=size;
}
}
void remove(int c)//删除节点c，以及c上下节点所在的行,每次调用这个函数，都是从列头节点开始向下删除
{
//这里c也可以理解为第c列 ,因为第c列的列头节点编号为c
L[R[c]]=L[c];
R[L[c]]=R[c];
for(int i=D[c]; i!=c; i=D[i])
for(int j=R[i]; j!=i; j=R[j])
{
U[D[j]]=U[j];
D[U[j]]=D[j];
--S[Col[j]];
}
}
void resume(int c)//恢复节点c,以及c上下节点所在的行(同上，也可以理解为从第c列的头节点开始恢复
{
for(int i=U[c]; i!=c; i=U[i])
for(int j=L[i]; j!=i; j=L[j])
++S[Col[U[D[j]]=D[U[j]]=j]]; //打这一行太纠结了 T T
L[R[c]]=R[L[c]]=c;
}
bool dance(int d)//递归深度
{
if(R[0]==0)
{
ansd=d;
return true;
}
int c=R[0];
for(int i=R[0]; i!=0; i=R[i])
if(S[i]<S[c])
c=i;
remove(c);//找到节点数最少的列，当前元素不是原图上0，1的节点，而是列头节点
for(int i=D[c]; i!=c; i=D[i])
{
ans[d]=Row[i];//列头节点下面的一个节点
for(int j=R[i]; j!=i; j=R[j])
remove(Col[j]);
if(dance(d+1))//找到，返回
return true;
for(int j=L[i]; j!=i; j=L[j])
resume(Col[j]);
}
resume(c);
return false;
}
} g;
int main()
{
int M,N;
while(scanf("%d%d",&M,&N)!=EOF)
{
g.init(M,N);
for(int i=1; i<=M; i++)
{
for(int j=1; j<=N; j++)
{
int num;
cin>>num;
if(num)
g.link(i,j);
}
}
if(g.dance(0)==false)
printf("It is impossible\n");
else
printf("Yes, I found it\n");
}
return 0;
}