Sudoku数独（dfs+状态压缩）

Sudoku数独

题目地址：https://ac.nowcoder.com/acm/problem/51013

题意

编程实现9x9数独

输入

将9x9的数独以字符串的形式输入，81个字符，'.'表示需要补齐的数字。
以‘end’作为结束符

输出

同输入的形式，81个数字字符。

题解

状态压缩：因为数独需要保证，每一行，每一列，每一宫的数字互不相同，那么就可以以二进制的形式表示这三种情况，比如：
举某个数独的一部分，第一行，第一列，第一个宫：
7 1 ? 4 ? 9 ? 5 8
3 ? 5
4 ? 8
9
?
1
?
6
8

第一行表示为000100110：即若有这个数字则标记为0，没有则标记为1，为了存储方便，化为十进制保存，即38
第一列表示为000010010
第一宫表示为100100010

因此只需要对这三种情况，“与”运算即可得到，都为1的位置，得到这个十进制数，然后通过lowbit得到依次得到，每一个1代表的数字，因此只需要记录所有的空值，然后通过位运算对所有空值dfs就可以得到最后答案。
因为数据保证有解，只需要输出满足数独的一种解即可，所以最终满足条件填满即可return，因此，采取先填哪些可能性少的空值，能避免过多次数的回溯，需要加上这一优化才能过这道题。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 12
#define maxm 100
#define INF 2147483640
#define IOS ios::sync_with_stdio(false)
#define mod 10007

struct node{
    int x,y;
};

int ff[N][N];
int R[N],C[N],G[N/3][N/3]; //状态压缩，1表示空值
vector<node>V;  //记录所有空值的坐标

int mp[1<<N],ones[1<<N];

int lowbit(int i)   //返回低位1
{
    return i&(-i);
}

int get_num(int x,int y)   //得到当前坐标，可以放那些数，返回状态压缩的值
{
    return R[x]&C[y]&G[x/3][y/3];
}

int get_one(int x)
{
    int cnt=0;
    while(x)
    {
        if(x%2==1)
            cnt++;
        x=x>>1;
    }
    return cnt;
}

bool dfs(int cur)
{
    if(cur==V.size())
        return true;

    int minv=12,id=0;
    for(int i=0;i<V.size();i++)
        if(ff[V[i].x][V[i].y]==0)
        {
            int t=ones[get_num(V[i].x,V[i].y)];
            if(t<minv)
            {
                minv=t;
                id=i;
            }
        }

    node tt=V[id];
    for(int i=get_num(tt.x,tt.y);i;i-=lowbit(i))
    {
        int va=lowbit(i);
        ff[tt.x][tt.y]=mp[va];
        R[tt.x]-=va;
        C[tt.y]-=va;
        G[tt.x/3][tt.y/3]-=va;
        if(dfs(cur+1))
            return true;
        R[tt.x]+=va;
        C[tt.y]+=va;
        G[tt.x/3][tt.y/3]+=va;
        ff[tt.x][tt.y]=0;
    }
    return false;
}

void init()
{
    V.clear();
    for(int i=0;i<N;i++)
    {
        R[i]=(1<<9)-1;
        C[i]=(1<<9)-1;
    }
    for(int i=0;i<N/3;i++)
        for(int j=0;j<N/3;j++)
            G[i][j]=(1<<9)-1;
}

int main()
{
    IOS;
    for(int i=0;i<9;i++)
        mp[1<<i]=i+1;
    for(int i=0;i<(1<<N);i++)
        ones[i]=get_one(i);

    string str;
    while(true)
    {
        cin>>str;
        if(str[0]=='e')
            break;

        init();
        for(int i=0; i<9; i++)
            for(int j=0; j<9; j++)
            {
                if(str[i*9+j]=='.')
                {
                    ff[i][j]=0;
                    V.push_back(node{i,j});
                }
                else
                {
                    ff[i][j]=str[i*9+j]-48;
                    int t=ff[i][j];
                    R[i]-=(1<<t-1);
                    C[j]-=(1<<t-1);
                    G[i/3][j/3]-=(1<<t-1);
                }
            }

        dfs(0);
        string ans="";
        for(int i=0; i<9; i++)
            for(int j=0; j<9; j++)
                ans.push_back(ff[i][j]+48);
        cout<<ans<<"\n";
    }
    return 0;
}

// https://www.nowcoder.com/practice/78a1a4ebe8a34c93aac006c44f6bf8a1?tpId=37&tqId=21267&rp=1&ru=/exam/oj/ta&qru=/exam/oj/ta&sourceUrl=%2Fexam%2Foj%2Fta%3Fpage%3D1%26pageSize%3D50%26search%3D%26tpId%3D37%26type%3D37&difficulty=5&judgeStatus=undefined&tags=&title=
#include <bits/stdc++.h>
using namespace std;

int ff[9][9];

int get_id(int x, int y) {
    // 块 id
    return x / 3 * 3 + y / 3;
}

bool heng[9][10], shu[9][10], kuai[9][10];
vector<pair<int,int>>V;
bool dfs(int p) {
    if(p==V.size())
        return true;
    int x=V[p].first, y=V[p].second;
    for(int j=1;j<=9;j++)
    {
        if(heng[x][j]==false&&shu[y][j]==false&&kuai[get_id(x,y)][j]==false)
        {
            heng[x][j]=true;
            shu[y][j]=true;
            kuai[get_id(x,y)][j]=true;
            ff[x][y]=j;
            if(dfs(p+1))
                return true;
            ff[x][y]=0;
            heng[x][j]=false;
            shu[y][j]=false;
            kuai[get_id(x,y)][j]=false;
        }
    }
    
    return false;
}

int main() {

    for (int i = 0; i < 9; i++)
        for (int j = 0; j < 9; j++)
        {
            cin >> ff[i][j];
            heng[i][ff[i][j]]=true;
            shu[j][ff[i][j]]=true;
            kuai[get_id(i, j)][ff[i][j]]=true;
            if(ff[i][j]==0)
                V.push_back(make_pair(i,j));
        }
            
    dfs(0);
    for (int i = 0; i < 9; i++) {
        for (int j = 0; j < 9; j++)
            cout<<ff[i][j]<<" ";
        cout << "\n";
    }

    return 0;
}
// 64 位输出请用 printf("%lld")spfa