Cod sursa(job #3261377)

Utilizator TofanCosminAlexandruTofan Cosmin TofanCosminAlexandru Data 5 decembrie 2024 17:28:11
Problema Problema Damelor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

ifstream in("damesah.in");
ofstream out("damesah.out");

int sol[14];
bool colOcup[28], mainDiagOcup[28], secDiagOcup[28];
int n, solCount;

bool isSafe(int i, int j) {
	return !(colOcup[j] || mainDiagOcup[n + j - i] || secDiagOcup[i + j]);
}

void printSol() {
	for (int i = 0; i < n; i++)
		out << sol[i] + 1 << " ";
	out << endl;
}

void solve(int i) {
	if (i == n) {
		if (solCount == 0)
			printSol();
		solCount++;
		return;
	}

	for (int j = 0; j < n; j++) {
		if (isSafe(i, j)) {
			sol[i] = j;
			colOcup[j] = mainDiagOcup[n + j - i] = secDiagOcup[i + j] = true;
			solve(i + 1);
			colOcup[j] = mainDiagOcup[n + j - i] = secDiagOcup[i + j] = false;
		}
	}
}

int main() {
	in >> n;
	solve(0);
	out << solCount << "\n";
}