我把之前 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。