题意就是题目描述了
小Ho最近遇到一个难题,他需要破解一个棋局。
棋局分成了n行,m列,每行有若干个棋子。小Ho需要从中选择若干行使得每一列有且恰好只有一个棋子。
用的也基本上是别人的代码 别人的注释也很详细 但是代码的很多小细节都想了很久
1.我们构造的是一个首位相接的十字链表 所以不用纠结col和H[row]到底是首元素还是尾元素
2.不用纠结col会不会和id冲突 我们很鸡贼的把id的初始值设为了m+1
(都想明白了之后又觉得没什么值得注意的了。。。)
#include <iostream>
using namespace std;
const int N=1e4 + 200;
const int Q=150;
int L[N],R[N],U[N],D[N];//上下左右的指向
int H[Q];//行头指针
int C[N],M[N];//元素所在行列
int S[Q];//每一列的元素数量
void link(int row,int col,int id)
{
C[id]=col;M[id]=row;
U[id]=U[col];D[id]=col;
D[U[col]]=id;U[col]=id;
if(H[row]==-1)
{
H[row]=L[id]=R[id]=id;
}
else
{
L[id]=L[H[row]];R[L[id]]=id;
L[H[row]]=id;R[id]=H[row];
}
S[col]++;
}
void remove(int col)
{
R[L[col]]=R[col];
L[R[col]]=L[col];
for(int i=D[col];i!=col;i=D[i])
{
for(int j=R[i];j!=i;j=R[j])
{
U[D[j]] = U[j];
D[U[j]] = D[j];
S[C[j]]--;
}
}
}
void resume(int col)
{
R[L[col]]=col;
L[R[col]]=col;
for(int i=U[col];i!=col;i=U[i])
{
for(int j=L[i];j!=i;j=L[j])
{
U[D[j]]=j;
D[U[j]]=j;
S[C[j]]++;
}
}
}
void init(int m)
{
for(int i=0;i<=m;i++)//初始化
{
L[i+1]=i;
R[i]=i+1;
U[i]=D[i] = i;
S[i]=0;
}
L[0]=m;
R[m]=0;
}
bool dance(int k)
{
if(!R[0])return true;
int col=R[0];
for(int i=R[0];i;i=R[i])
if(S[i]<S[col])
col=i;
remove(col);
for(int i=D[col];i!=col;i=D[i])
{
for(int j=R[i];j!=i;j=R[j])
remove(C[j]);
if(dance(k+1))return true;
for(int j=L[i];j!=i;j=L[j])
resume(C[j]);
}
resume(col);
return false;
}
int main()
{
int T,n,m,x;
cin>>T;
while(T--)
{
cin>>n>>m;
init(m);
int id=m+1;
for(int i=1;i<=n;i++)
{
H[i]=-1;
for(int j=1;j<=m;j++)
{
cin>>x;
if(x)
link(i,j,id++);
}
}
cout<< (dance(0)?"Yes":"No") <<endl;
}
return 0;
}