poj3074 9*9数独,poj3076 16*16数独 DLX

     两题类似,都是求数独,用DLX解数独最大的问题就是建表..对9*9的数独,建立的是(9*9*9)*(9*9*4)的01矩阵去做的。行表示9*9个格子中,每个格子的9种填法,列的话,得有个9*9表示数字放在第i行第j列,第二个9*9表示在第i行放的数是j,第三个9*9表示在第i列放的数是j,第四个9*9表示在第i个九宫格放的数是j.读取数据的时候,如果某个位置上有数字的话,那么添加一行就可以了(已有的数可以看做这个位置只有这一种选择),如果某个位置是空,那么就添加9行,表示1--9的9种填法,建好表以后直接跑一遍DLX就可以了。16*16的数独也是同样的做法。

poj3074

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1000*1000;
const int inf=(1<<29);
const int ct[]={1,82,163,244};

int n,m;
int S[1010],st[1010];
int L[maxn],R[maxn],U[maxn],D[maxn];
int col[maxn],row[maxn];
struct ANS
{
    int r,c,k;
}ans[maxn];
int M,K;
char ss[120];
struct DLX
{
    void remove(int 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)
    {
        L[R[c]]=c;
        R[L[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]];
            }
        }
    }
    bool dfs(int k)
    {
        if (R[0]==0)
        {
            memset(ss,0,sizeof ss);
            for (int i=0; i<k; i++)
            ss[ans[st[i]].r*9+ans[st[i]].c]=ans[st[i]].k+'0';
            puts(ss);
            return true;
        }
        int ts=inf,c=R[0];
        for (int i=R[c]; i!=0; i=R[i])
        if (S[c]>S[i]) c=i;

        remove(c);
        for (int i=D[c]; i!=c; i=D[i])
        {
            st[k]=row[i];
            for (int j=R[i]; j!=i; j=R[j]) remove(col[j]);
            if (dfs(k+1)) return true;
            for (int j=L[i]; j!=i; j=L[j]) resume(col[j]);
        }
        resume(c);
        return false;
    }

    void init(int m)
    {

        for (int i=1; i<=m; i++)
        {
            L[i]=i-1;
            R[i]=i+1;
            U[i]=D[i]=i;
            col[i]=i;
            row[i]=0;
            S[i]=0;
        }
        L[0]=m;
        R[0]=1;
        U[0]=D[0]=0;
        R[m]=0;

        K=m;
        M=0;
    }
    int makehead(int c)
    {
        K++;
        S[c]++;
        col[K]=c;
        row[K]=M;
        L[K]=R[K]=K;

        U[K]=c;
        D[K]=D[c];
        U[D[K]]=K;
        D[U[K]]=K;
        return K;
    }
    void addcol(int id,int c)
    {
        K++;
        S[c]++;
        col[K]=c;
        row[K]=M;

        L[K]=id;
        R[K]=R[id];
        L[R[K]]=R[L[K]]=K;

        U[K]=c;
        D[K]=D[c];
        U[D[K]]=D[U[K]]=K;
    }

    void addrow(int i,int j,int k)
    {
       int id;
       M++;
       ans[M].r=i;
       ans[M].c=j;
       ans[M].k=k+1;
       id=makehead(9*i+j+ct[0]);
       addcol(id,9*i+k+ct[1]);
       addcol(id,9*j+k+ct[2]);
       addcol(id,(i/3*3+j/3)*9+k+ct[3]);
    }
}dlx;
char rs[200];
int main()
{
//    freopen("in.txt","r",stdin);
    while(gets(rs))
    {
        if (rs[0]=='e') break;
        dlx.init(9*9*4);
        for (int i=0; i<9; i++)
         for (int j=0; j<9; j++)
         {
             if (rs[i*9+j]=='.')
             {
                for (int k=0; k<9; k++)
                dlx.addrow(i,j,k);
             }
             else
             {
                dlx.addrow(i,j,rs[i*9+j]-'1');
             }
         }
         dlx.dfs(0);


    }
    return 0;
}


poj3076

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=3000*3000;
const int inf=(1<<29);
const int ct[]={1,16*16+1,16*16*2+1,16*16*3+1};

int n,m;
int S[16*16*16],st[16*16*16];
int L[maxn],R[maxn],U[maxn],D[maxn];
int col[maxn],row[maxn];
struct ANS
{
    int r,c,k;
}ans[maxn];
int M,K;
char ss[520];
struct DLX
{
    void remove(int 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)
    {
        L[R[c]]=c;
        R[L[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]];
            }
        }
    }
    bool dfs(int k)
    {
        if (R[0]==0)
        {
            memset(ss,0,sizeof ss);
            for (int i=0; i<k; i++)
            ss[ans[st[i]].r*16+ans[st[i]].c]=ans[st[i]].k+'A';
            for (int i=0; i<16; i++)
            {
                for (int j=0; j<16; j++)
                cout<<ss[i*16+j];
                cout<<endl;
            }
            return true;
        }
        int ts=inf,c=R[0];
        for (int i=R[c]; i!=0; i=R[i])
        if (S[c]>S[i]) c=i;

        remove(c);
        for (int i=D[c]; i!=c; i=D[i])
        {
            st[k]=row[i];
            for (int j=R[i]; j!=i; j=R[j]) remove(col[j]);
            if (dfs(k+1)) return true;
            for (int j=L[i]; j!=i; j=L[j]) resume(col[j]);
        }
        resume(c);
        return false;
    }

    void init(int m)
    {

        for (int i=1; i<=m; i++)
        {
            L[i]=i-1;
            R[i]=i+1;
            U[i]=D[i]=i;
            col[i]=i;
            row[i]=0;
            S[i]=0;
        }
        L[0]=m;
        R[0]=1;
        U[0]=D[0]=0;
        R[m]=0;

        K=m;
        M=0;
    }
    int makehead(int c)
    {
        K++;
        S[c]++;
        col[K]=c;
        row[K]=M;
        L[K]=R[K]=K;

        U[K]=c;
        D[K]=D[c];
        U[D[K]]=K;
        D[U[K]]=K;
        return K;
    }
    void addcol(int id,int c)
    {
        K++;
        S[c]++;
        col[K]=c;
        row[K]=M;

        L[K]=id;
        R[K]=R[id];
        L[R[K]]=R[L[K]]=K;

        U[K]=c;
        D[K]=D[c];
        U[D[K]]=D[U[K]]=K;
    }

    void addrow(int i,int j,int k)
    {
       int id;
       M++;
       ans[M].r=i;
       ans[M].c=j;
       ans[M].k=k;
       id=makehead(16*i+j+ct[0]);
       addcol(id,16*i+k+ct[1]);
       addcol(id,16*j+k+ct[2]);
       addcol(id,(i/4*4+j/4)*16+k+ct[3]);
    }
}dlx;
char rs[30][30];
int main()
{
//    freopen("in.txt","r",stdin);
    while(cin>>rs[0])
    {
        for (int i=1; i<=15; i++) cin>>rs[i];

        dlx.init(16*16*4);
        for (int i=0; i<16; i++)
         for (int j=0; j<16; j++)
         {
             if (rs[i][j]=='-')
             {
                for (int k=0; k<16; k++)
                dlx.addrow(i,j,k);
             }
             else
             {
                dlx.addrow(i,j,rs[i][j]-'A');
             }
         }
         dlx.dfs(0);
        cout<<endl;

    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这是一道比较经典的计数问题。题目描述如下: 给定一个 $n \times n$ 的网格图,其中一些格子被标记为障碍。一个连通块是指一些被标记为障碍的格子的集合,满足这些格子在网格图中连通。一个格子是连通的当且仅当它与另一个被标记为障碍的格子在网格图中有公共边。 现在,你需要计算在这个网格图中,有多少个不同的连通块,满足这个连通块的大小(即包含的格子数)恰好为 $k$。 这是一道比较经典的计数问题,一般可以通过计算生成函数的方法来解决。具体来说,我们可以定义一个生成函数 $F(x)$,其中 $[x^k]F(x)$ 表示大小为 $k$ 的连通块的个数。那么,我们可以考虑如何计算这个生成函数。 对于一个大小为 $k$ 的连通块,我们可以考虑它的形状。具体来说,我们可以考虑以该连通块的最左边、最上边的格子为起点,从上到下、从左到右遍历该连通块,把每个格子在该连通块中的相对位置记录下来。由于该连通块的大小为 $k$,因此这些相对位置一定是 $(x,y) \in [0,n-1]^2$ 中的 $k$ 个不同点。 现在,我们需要考虑如何计算这些点对应的连通块是否合法。具体来说,我们可以考虑从左到右、从上到下依次处理这些点,对于每个点 $(x,y)$,我们需要考虑它是否能够与左边的点和上边的点连通。具体来说,如果 $(x-1,y)$ 和 $(x,y)$ 都在该连通块中且它们在网格图中有公共边,那么它们就是连通的;同样,如果 $(x,y-1)$ 和 $(x,y)$ 都在该连通块中且它们在网格图中有公共边,那么它们也是连通的。如果 $(x,y)$ 与左边和上边的点都不连通,那么说明这个点不属于该连通块。 考虑到每个点最多只有两个方向需要检查,因此时间复杂度为 $O(n^2 k)$。不过,我们可以使用类似于矩阵乘法的思想,将这个过程优化到 $O(k^3)$ 的时间复杂度。 具体来说,我们可以设 $f_{i,j,k}$ 表示状态 $(i,j)$ 所代表的点在连通块中,且连通块的大小为 $k$ 的方案数。显然,对于一个合法的 $(i,j,k)$,我们可以考虑 $(i-1,j,k-1)$ 和 $(i,j-1,k-1)$ 这两个状态,然后把点 $(i,j)$ 加入到它们所代表的连通块中。因此,我们可以设计一个 $O(k^3)$ 的 DP 状态转移,计算 $f_{i,j,k}$。 具体来说,我们可以考虑枚举连通块所包含的最右边和最下边的格子的坐标 $(x,y)$,然后计算 $f_{x,y,k}$。对于一个合法的 $(x,y,k)$,我们可以考虑将 $(x,y)$ 所代表的点加入到 $(x-1,y,k-1)$ 和 $(x,y-1,k-1)$ 所代表的连通块中。不过,这里需要注意一个细节:如果 $(x-1,y)$ 和 $(x,y)$ 在网格图中没有相邻边,那么它们不能算作连通的。因此,我们需要特判这个情况。 最终,$f_{n,n,k}$ 就是大小为 $k$ 的连通块的个数,时间复杂度为 $O(n^2 k + k^3)$。 参考代码:

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值