Cod sursa(job #3145590)

Utilizator victor_gabrielVictor Tene victor_gabriel Data 16 august 2023 12:51:00
Problema Problema Damelor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <fstream>
#include <map>

using namespace std;

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

const int DIM = 15;

int n, solCnt;
int sol[DIM];
bool isUsed[DIM], mainDiag[2 * DIM], secDiag[2 * DIM];

void backtrack(int step) {
    if (step == n + 1) {
        solCnt++;
        if (solCnt == 1) {
            for (int i = 1; i <= n; i++)
                fout << sol[i] << ' ';
            fout << '\n';
        }
    } else {
        for (int i = 1; i <= n; i++) {
            if (!isUsed[i] && !mainDiag[n + step - i] && !secDiag[step + i]) {
                sol[step] = i;
                isUsed[i] = mainDiag[n + step - i] = secDiag[step + i] = true;
                backtrack(step + 1);
                isUsed[i] = mainDiag[n + step - i] = secDiag[step + i] = false;
            }
        }
    }
}

int main() {
    fin >> n;

    solCnt = 0;
    backtrack(1);
    fout << solCnt;

    fin.close();
    fout.close();
    return 0;
}