5 October

POJ2676 Sudoku

位运算 + 搜索。更好的优化方法：方案数最小的空格先填。

把某一位 置为 0：a &=~ (1<<n)

把某一位 置为 1：a |= (1<<n)

#include <cstdio>
#include <cstring>
int T, f[9], g[9], h[3][3];
char buf[10]; 
int v[9][9], tot;

inline int lowbit(int x) { return x&(-x); }
inline int num(int x) {
	register int cnt=0; while (x) ++cnt, x>>=1; return cnt-1;
}

bool search(int x, int y) {
	int res=f[x]&g[y]&h[x/3][y/3];
	if (!res) return 0;
 	while (res) {
		int t=lowbit(res); res&=~t; --tot;
		f[x]&=~t, g[y]&=~t, h[x/3][y/3]&=~t; v[x][y]=num(t);
		if (!tot) return 1;
		for (int i=0, bk=0; i<9; ++i) {
			for (int j=0; j<9; ++j)
				if (!v[i][j]) {if (search(i, j)) return 1; bk=1; break; }
			if (bk) break;
		}
		f[x]|=t, g[y]|=t, h[x/3][y/3]|=t; v[x][y]=0; ++tot;
	}
	return 0;
}

int main() {
	scanf("%d", &T);
	while (T--) {
		for (int i=0; i<9; ++i) for (int j=0; j<9; ++j)
			f[i]=g[i]=(1<<10)-2;
		for (int i=0; i<3; ++i) for (int j=0; j<3; ++j) 
			h[i][j]=(1<<10)-2;
		memset(v, 0, sizeof v); tot=0;
		for (int i=0; i<9; ++i) {
			scanf("%s", buf);
			for (int j=0; j<9; ++j) if (buf[j]-='0') {
				v[i][j]=buf[j];
				f[i]&=~(1<<buf[j]), g[j]&=~(1<<buf[j]),
				h[i/3][j/3]&=~(1<<buf[j]);
			} else ++tot;
		}
		for (int i=0, bk=0; i<9; ++i) {
			for (int j=0; j<9; ++j)
				if (!v[i][j]) {search(i, j); bk=1; break; }
			if (bk) break;
		}
		for (int i=0; i<9; ++i) {
			for (int j=0; j<9; ++j) printf("%d", v[i][j]);
			printf("
");
		}
	}
	return 0;
}