洛谷 lP1784 数独 —— 舞蹈链(DLX)精确覆盖

This way

题意:

在这里插入图片描述
宫是指一个九宫格

题解:

要求是每一个点只能出现一个值,每一行,每一列,每一宫都要出现1~9里面的所有数。那么每个格子是所有数都互斥的,而行,列,宫出现情况是相同数互斥,不同数互补的,我们就以每个格子出现的情况为行,是9*9*9种情况,以i*9*9+j*9+k(i是行,j是列,k是这个数的值)来唯一标识第i行第j列放了数k。
那么这个数放了之后,这一个格子就不能再放数,这一行,列,宫就不能再放k。将这些状态放在第i*9*9+j*9+k行的四个列上。
接下来i和j都是从0到8,因为精确覆盖的话,要求每一列都要有正好一个1,那么下标必须得从1开始,中间的区间必须是连续的(如果你魔改了,当我没说)
我们以i*9+j+1表示这个格子,i*9+81+k表示第i行放了k,j*9+81*2+k表示第j列放了k,(i/3*3+j/3)*9+81*3+k表示这一宫放了k
那么行数是9*9*9,列数是81*4

#include<bits/stdc++.h>
using namespace std;
const int N=350*1000;
int n=729,m=324;//要求n,m非常精准
int cnt;
int l[N],r[N],u[N],d[N],col[N],row[N];//每个点的左右上下指针,所在行列
int h[N];//每行的头结点
int s[N];//每列的节点数
int ansk[N];//选了那些集合
void init(int m){//每行m个元素
    for(int i=0;i<=m;i++){
        r[i]=i+1;
        l[i]=i-1;
        u[i]=d[i]=i;
    }
    r[m]=0;
    l[0]=m;
    memset(h,-1,sizeof(h));
    memset(s,0,sizeof(s));
    cnt=m+1;
}//初始化
void add(int R,int C){//R行C列插入点
    s[C]++;
    row[cnt]=R;
    col[cnt]=C;
    u[cnt]=C;
    d[cnt]=d[C];
    u[d[C]]=cnt;
    d[C]=cnt;
    if(h[R]==-1)h[R]=r[cnt]=l[cnt]=cnt;//该行没有点,直接加入
    else{
        r[cnt]=h[R];
        l[cnt]=l[h[R]];
        r[l[h[R]]]=cnt;
        l[h[R]]=cnt;
    }
    cnt++;
    //return;
}
void remove(int C){//删除涉及C列的集合
    r[l[C]]=r[C],l[r[C]]=l[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列的集合
    for(int i=u[C];i!=C;i=u[i]){
        for(int j=l[i];j!=i;j=l[j]){
            u[d[j]]=j;
            d[u[j]]=j;
            s[col[j]]++;
        }
    }
    r[l[C]]=C;
    l[r[C]]=C;
}
int a[10][10];
bool dance(int deep){//deep表示已经选了多少行
    if(r[0]==0){//表示整个矩阵删完了
        int i=0;
        for(i=0;i<deep;i++)
            a[(ansk[i]-1)/9/9][(ansk[i]-1)/9%9]=ansk[i]%9==0?9:ansk[i]%9;
        return 1;
    }
    int c=r[0];
    int i,j;
    for(i=r[0];i!=0;i=r[i])if(s[i]<s[c])c=i;
    remove(c);
    for(i=d[c];i!=c;i=d[i]){
        ansk[deep]=row[i];
        for(j=r[i];j!=i;j=r[j])remove(col[j]);
        if(dance(deep+1))return 1;
        for(j=l[i];j!=i;j=l[j])resume(col[j]);
    }
    resume(c);
    return 0;
}
int main()
{
    init(m);
    for(int i=0;i<=8;i++){
        for(int j=0;j<=8;j++){
            int x;
            scanf("%d",&x);
            for(int k=1;k<=9;k++){
                if(x!=0&&k!=x)continue;
                int cas=i*9*9+j*9+k;
                add(cas,i*9+j+1);//这个格子有了
                add(cas,i*9+81+k);//这一行有了k
                add(cas,j*9+81*2+k);//这一列有了k
                add(cas,(i/3*3+j/3)*9+81*3+k);//这一宫有了k
            }
        }
    }
    dance(0);
    for(int i=0;i<=8;i++){
        for(int j=0;j<=8;j++)
            printf("%d%c",a[i][j]," \n"[i==n]);
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/107653520