Cod sursa(job #3187219)

Utilizator Bogdan345Marius Mihalache Bogdan345 Data 28 decembrie 2023 10:09:48
Problema Problema Damelor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.42 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;
bool [30] folosit;
bool [30] diagonala1;
bool [30]diagonala2;
vector<int> rasp;
int cont;
bool ok;

void backtracking(int n, int linie, ofstream& f) {
    if (linie == n + 1) {
        if (ok == false) {
            for (int i = 1; i <= n; i++) {
                f << rasp[i] << " ";
            }
            f << endl;
            cont = 1;
            ok = true;
            return;
        } else {
            cont += 1;
            return;
        }
    }
    for (int coloana = 1; coloana <= n; coloana++) {
        if (diagonala1[linie - coloana + n] == true || diagonala2[linie + coloana] == true || folosit[coloana] == true) {
            continue;
        }
        diagonala1[linie - coloana + n] = true;
        diagonala2[linie + coloana] = true;
        folosit[coloana] = true;
        rasp[linie] = coloana;
        backtracking(n, linie + 1, f);
        diagonala1[linie - coloana + n] = false;
        diagonala2[linie + coloana] = false;
        folosit[coloana] = false;
    }
}

int main() {
    ifstream g("damesah.in");
    ofstream f("damesah.out");
    int n;
    g >> n;
    folosit.resize(30, false);
    diagonala1.resize(30, false);
    diagonala2.resize(30, false);
    rasp.resize(n + 1, 0);
    cont = 0;
    ok = false;
    backtracking(n, 1, f);
    f << cont;
    return 0;
}