数独DFS &&位运算

DFS &&位运算

#include<cstdio>
#include<algorithm>
#include<set>
#include<iostream>
using namespace std;
const int N=9;
int r[N],col[N],cell[3][3];
char str[100];
int ones[1<<N],map[1<<N];
inline lowbit(int x)//返回x的最后一位1 
{
    return (-x)&x;
}
void init()//初始化 
{
    for(int i=0;i<N;i++)
    {
        r[i]=col[i]=(1<<N)-1;//初始化所有的都可取 
    }
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
        {
            cell[i][j]=(1<<N)-1;//初始化九宫格的每一格的所有数字都可以取 
        }
    }
}
inline int get(int x,int y)
{
    return r[x]&col[y]&cell[x/3][y/3];//通过行列九宫格判断该位置可能取的值 
}
bool dfs(int cnt)
{
    int x,y;
    if(!cnt) return true;
    int minn=10;//寻找备选数最少的点 
    for(int i=0;i<N;i++)
    {
        for(int j=0;j<N;j++)
        {
            if(str[i*9+j]=='.')
            {
                int t=ones[get(i,j)]; //对应的可能取得值的个数 
                if(t<minn)
                {
                    minn=t;
                    x=i;
                    y=j;
                    
                }
            }
        }
    }
 for(int i=get(x,y);i;i-=lowbit(i))
 {
     int t=map[lowbit(i)];
     r[x]-=1<<t;
     col[y]-=1<<t;
     cell[x/3][y/3]-=1<<t;
     str[x*9+y]='1'+t;
     if(dfs(cnt-1)) return true;
     //恢复原来的情况 
      r[x]+=1<<t;
     col[y]+=1<<t;
     cell[x/3][y/3]+=1<<t;
     str[x*9+y]='.';
 }
 return false;
}
int main()
{
    for(int i=0;i<N;i++)
    {
        map[1<<i]=i;//键值是其代表的数 
    }
    for(int i=0;i<1<<N;i++)
    {
        int s=0;
        for(int j=i;j;j-=lowbit(j))
        {
            s++;
        }
        ones[i]=s;//i的二进制中有s个1
    }
    while(cin>>str,str[0]!='e')
    {
        init();
        int cnt=0;
        for(int i=0,k=0;i<N;i++)
        {
            for(int j=0;j<N;j++,k++)
            {
                if(str[k]!='.')
                {
                    int t=str[k]-'1';//说明在该行该列以及这个九宫格中这个数是不可以取得的 
                    r[i]-=1<<t; 
                    col[j]-=1<<t;
                    cell[i/3][j/3]-=1<<t;
                }else 
                {
                    cnt++;//未知点++； 
                }
                
            }
        }
        dfs(cnt);
        cout<<str<<endl;
    }
    return 0;
}

#include<iostream>
#include<algorithm>
using namespace std;
const int N=9;
int c[N],r[N],cell[3][3],one[1<<N],mp[1<<N];
char str[100];
inline int lowbit(int x)
{
    return (-x)&x;
}
void init()
{
    for(int i=0;i<N;i++)
    {
        c[i]=(1<<N)-1;
        r[i]=(1<<N)-1;
    }
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
          cell[i][j]=(1<<N)-1;
    }
}
inline int get(int x,int y)
{
    return r[x]&c[y]&cell[x/3][y/3];
}
bool dfs(int cnt)
{
    if(cnt==0) return true;
    int x,y,minn=10;
    for(int i=0;i<N;i++)
    {
        for(int j=0;j<N;j++)
        {
            if(str[i*9+j]=='.')
            {
                int t=one[get(i,j)];
                if(t<minn)
                {
                    x=i;
                    y=j;
                    minn=t;
                }
            }
        }
    }
    for(int i=get(x,y);i;i-=lowbit(i))
    {
        int t=mp[lowbit(i)];
        r[x]-=1<<t;
        c[y]-=1<<t;
        cell[x/3][y/3]-=1<<t;
        str[x*9+y]='1'+t;
        if(dfs(cnt-1)) return true;
        r[x]+=1<<t;
        c[y]+=1<<t;
        cell[x/3][y/3]+=1<<t;
        str[x*9+y]='.';
    }
    return false;
}
int main()
{
    for(int i=0;i<N;i++)
    {
        mp[1<<i]=i;//键值是其代表的数 
    }
    for(int i=0;i<1<<N;i++)
    {
        int s=0;
        for(int j=i;j;j-=lowbit(j))
        {
            s++;
        }
        one[i]=s;//i的二进制中有s个1
    }
    while(cin>>str,str[0]!='e')
    {
        init();
        int cnt=0;
        for(int i=0,k=0;i<N;i++)
        {
            for(int j=0;j<N;j++,k++)
            {
                if(str[k]!='.')
                {
                    int t=str[k]-'1';//说明在该行该列以及这个九宫格中这个数是不可以取得的 
                    r[i]-=1<<t; 
                    c[j]-=1<<t;
                    cell[i/3][j/3]-=1<<t;
                }else 
                {
                    cnt++;//未知点++； 
                }
                
            }
        }
        dfs(cnt);
        cout<<str<<endl;
    }
    return 0;
}