DLX pku 3076

标准数独,精确覆盖
// pku3076.cpp
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#define out(v) cout << #v << ": " << (v) << endl
using namespace std;
typedef long long LL;

const int maxN = 20 * 20 * 20, maxM = 4 * 20 * 20, N = 16, NN = N * N, n = 4;
const int max_size = maxN * maxM;
const int inf = 0x3f3f3f3f;
int L[max_size], R[max_size], U[max_size], D[max_size], CH[max_size], RH[max_size];
int S[maxM], O[maxM];
int head, size;

int node(int up, int down, int left, int right) {
	U[size] = up, D[size] = down;
	L[size] = left, R[size] = right;
	D[up] = U[down] = R[left] = L[right] = size;
	return size++;
}
bool mat[maxN][maxM];
void init(int N, int M) {
	size = 0;
	head = node(0, 0, 0, 0);
	for (int j = 1; j <= M; ++j) {
		CH[j] = node(size, size, L[head], head), S[j] = 0;
	}
	for (int i = 1; i <= N; ++i) {
		int row = -1, k;
		for (int j = 1; j <= M; ++j) {
			if (!mat[i][j]) continue;
			if (row == -1) {
				row = node(U[CH[j]], CH[j], size, size);
				RH[row] = i, CH[row] = CH[j], ++S[j];
			} else {
				k = node(U[CH[j]], CH[j], L[row], row);
				RH[k] = i, CH[k] = CH[j], ++S[j];
			}
		}
	}
}

void remove(const int &c) {
	L[R[c]] = L[c], R[L[c]] = R[c];
	for (int i = D[c]; i != c; i = D[i]) {
		for (int j = R[i]; j != i; j = R[j]) {
			U[D[j]] = U[j], D[U[j]] = D[j];
			--S[CH[j]];
		}
	}
}
void resume(const int &c) {
	for (int i = U[c]; i != c; i = U[i]) {
		for (int j = L[i]; j != i; j = L[j]) {
			++S[CH[j]];
			U[D[j]] = D[U[j]] = j;
		}
	}
	L[R[c]] = R[L[c]] = c;
}
int len;
bool dfs(const int &k) {
	if (R[head] == head) {
		len = k - 1;
		return true;
	}
	int s = inf, c;
	for (int t = R[head]; t != head; t = R[t]) {
		if (S[t] < s) s = S[t], c = t;
	}
	remove(c);
	for (int i = D[c]; i != c; i = D[i]) {
		O[k] = RH[i];
		for (int j = R[i]; j != i; j = R[j]) {
			remove(CH[j]);
		}
		if (dfs(k + 1)) {
			return true;
		}
		for (int j = L[i]; j != i; j = L[j]) {
			resume(CH[j]);
		}
	}
	resume(c);
	return false;
}

char sudoku[20][20];
void make()
{
	memset(mat, false, sizeof(mat));
	const int N = 16, NN = N * N, n = 4;
	for (int i = 1; i <= N; ++i)
		for (int j = 1; j <= N; ++j)
			for (int k = 1; k <= N; ++k)
				if (sudoku[i][j] == '-' || sudoku[i][j] == 'A' + k - 1)
					mat[(i - 1) * NN + (j - 1) * N + k][(i - 1) * N + j] = true;
	for (int i = 1; i <= N; ++i)
		for (int k = 1; k <= N; ++k)
			for (int j = 1; j <= N; ++j)
				if (sudoku[i][j] == '-' || sudoku[i][j] == 'A' + k - 1)
					mat[(i - 1) * NN + (j - 1) * N + k][NN + (i - 1) * N + k] = true;
	for (int j = 1; j <= N; ++j)
		for (int k = 1; k <= N; ++k)
			for (int i = 1; i <= N; ++i)
				if (sudoku[i][j] == '-' || sudoku[i][j] == 'A' + k - 1)
					mat[(i - 1) * NN + (j - 1) * N + k][NN * 2 + (j - 1) * N + k] = true;
	int region;
	for (int i = 1; i <= N; ++i)
		for (int j = 1; j <= N; ++j)
			for (int k = 1; k <= N; ++k) {
				region = ((i - 1) / n) * n + (j - 1) / n + 1;
				if (sudoku[i][j] == '-' || sudoku[i][j] == 'A' + k - 1)
					mat[(i - 1) * NN + (j - 1) * N + k][NN * 3 + (region - 1) * N + k] = true;
			}
}

int main()
{
	bool first = true;
	while (scanf("%s", sudoku[1] + 1) != EOF)
	{
		for (int i = 2; i <= N; ++i)
			scanf("%s", sudoku[i] + 1);
		make();
		init(N * N * N, 4 * N * N);
		dfs(1);
		for (int i = 1; i <= len; ++i) {
			int x = (O[i] - 1) / NN + 1;
			O[i] -= (x - 1) * NN;
			int y = (O[i] - 1) / N + 1;
			O[i] -= (y - 1) * N;
			int z = O[i];
			sudoku[x][y] = 'A' + z - 1;
		}
		if (first) first = false;
		else {
			putchar('\n');
		}
		for (int i = 1; i <= N; ++i) {
			for (int j = 1; j <= N; ++j)
				putchar(sudoku[i][j]);
			putchar('\n');
		}
	}
	return 0;
}

你可能感兴趣的:(C++,c,C#,J#,UP)