hihoCoder #1317 : 搜索四·跳舞链 （精确覆盖）

题意就是题目描述了

小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;
}