ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

数独

2022-09-02 15:02:26  阅读:113  来源: 互联网

标签:map int ++ ones str include 数独


https://www.acwing.com/problem/content/168/

#include <algorithm>
#include <iostream>

using namespace std;

const int N = 9;

// 0- 511
// ones[i] 表示i这个数字的二进制表示中, 有多少个1
// map快速求出log_2[x], 例如: map[8] = 3;
int  ones[1 << N], map[1 << N];
int  row[N], col[N], cell[3][3];
char str[100];

inline int lowbit(int x)
{
    return x & -x;
}

void init()
{
    for (int i = 0; i < N; i++) {
        row[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 row[x] & col[y] & cell[x / 3][y / 3];
}

bool dfs(int cnt)
{
    if (!cnt) {
        return true;
    }

    // 找出可选方案最少的位置
    int minv = 10;
    int x, y;  //可能方案最少的位置的横纵坐标
    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 < minv) {
                    minv = t;
                    x = i, y = j;
                }
            }
        }
    }

    for (int i = get(x, y); i; i -= lowbit(i)) {
        int t = map[lowbit(i)];

        //修改状态
        row[x] -= 1 << t;
        col[y] -= 1 << t;
        cell[x / 3][y / 3] -= 1 << t;
        str[x * 9 + y] = '1' + t;  //二维变一维

        //如果能成功, 那么就返回true
        if (dfs(cnt - 1)) {
            return true;
        }

        //恢复现场
        row[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';
                    row[i] -= 1 << t;
                    col[j] -= 1 << t;
                    cell[i / 3][j / 3] -= 1 << t;
                }
                else {
                    cnt++;
                }
            }
        }
        dfs(cnt);
        cout << str << endl;
    }
    return 0;
}

标签:map,int,++,ones,str,include,数独
来源: https://www.cnblogs.com/xjtfate/p/16649888.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有