16宫数独求助
  • 板块题目总版
  • 楼主__K2FeO4
  • 当前回复2
  • 已保存回复2
  • 发布时间2022/9/12 10:13
  • 上次更新2023/10/27 11:54:29
查看原帖
16宫数独求助
565450
__K2FeO4楼主2022/9/12 10:13

我把之前 9 宫格的代码循环次数、数组大小改了一下,在本地跑 TLE。

我 Baidu First Search 了之后,使用了优化版本,循环次数、数组大小改后成了 RE。我知道是 lowbit() 出现了越界问题,但不知道怎么改。

#include<bits/stdc++.h>
using namespace std;
const int N = 16, M = 1 << N;        // M=65536
int ones[M+4], mp[M+4];// 快速统计每个状态有多少个 1,lowbit() 操作返回 2^a,求这个 a 等价于求 log
int row[N+4], col[N+4], cell[4][4];
int str[260];
int lowbit(int x) {
    return x & -x;
}
int get(int x, int y) {
    return row[x] & col[y] & cell[x / 4][y / 4];
}
// 0 代表该数已使用,1 代表该数未使用
void init() {
    for (int i = 0; i < N; i ++ ) row[i] = col[i] = (1 << N) - 1;
    for (int i = 0; i < 4; i ++ )
        for (int j = 0; j < 4; j ++ )
            cell[i][j] = (1 << N) - 1;
}
// 将 (x, y) 位置填入数字 t,is_set 用来标记是填数字还是恢复现场这两种操作
// 可用于初始化、填数字、回溯三种,用一个函数封装,属实流畅!
void draw(int x, int y, int t, bool is_set) {
    if (is_set) str[x * N + y] = t;
    else str[x * N + y] = 0;

    // 注意 0 代表数已使用,1 代表该数未使用,状态得减....
    int v = 1 << t;
    if (!is_set) v = -v;

    row[x] -= v;
    col[y] -= v;
    cell[x / 4][y / 4] -= v;
}
bool dfs(int cnt) {
	//printf("%d\n",cnt);
    if (!cnt) return true;
    int minv = 17;
    int x, y;
    for (int i = 0, k = 0; i < N; i ++ ) 
    for (int j = 0; j < N; j ++ , k ++ ){
        if (!str[k]){
            int state = get(i, j);
            if (ones[state] < minv){
                minv = ones[state];
                x = i, y = j;
            }
        }
    }
    int state = get(x, y);
    ///printf("%d ",state);
	for (int i = state; i; i -= lowbit(i)) {
        int t = mp[lowbit(i)];                      // 找到最后一位 1 的位置
        draw(x, y, t, true);
        if (dfs(cnt - 1)) return true;
        draw(x, y, t, false);
    }

    return false;
}
int main() {
    for (int i = 0; i < N; i ++ ) mp[1 << i] = i;
    for (int i = 0; i < 1 << N; i ++ ) {
        int t = i, cnt = 0;
        while (t) t -= lowbit(t), cnt ++ ;
        ones[i] = cnt;
    }
    for (int i = 0; i < 256; i ++ ) scanf("%d", str + i);
    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];       // 初始化数字情况
                draw(i, j, t, true);
            } else {
                cnt ++ ;
            }
    dfs(cnt);
    for (int i = 0; i < 256; i += 16) {
    	for (int j = 0; j < 16; j++)
    	printf("%d ", str[i + j]);
    	puts("");
	}
    return 0;
}

我自己出的题

时限最多开到 5s,最后一个点就 T 了。

以上代码无法过样例,会 RE。

2022/9/12 10:13
加载中...