Cod sursa(job #1991699)

Utilizator MaligMamaliga cu smantana Malig Data 18 iunie 2017 08:52:24
Problema Problema Damelor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include <iostream>
#include <fstream>

using namespace std;
ifstream in("damesah.in");
ofstream out("damesah.out");

#define ll long long
#define pb push_back
const int inf = 1e9 + 5;
const int NMax = 15;

int N,nrSol;
int queen[NMax];
bool col[NMax],diagPrin[2*NMax],diagSec[2*NMax];

void backT(int);

int main() {
    in>>N;

    backT(1);
    out<<nrSol<<'\n';

    in.close();out.close();
    return 0;
}

void backT(int i) {
    if (i == N+1) {
        if (!nrSol) {
            for (int k=1;k <= N;++k) {
                out<<queen[k]<<' ';
            }
            out<<'\n';
        }

        ++nrSol;
        return;
    }

    for (int j=1;j <= N;++j) {
        if (!col[j] && !diagPrin[N + i-j] && !diagSec[i+j]) {

            col[j] = diagPrin[N + i-j] = diagSec[i+j] = true;
            queen[i] = j;
            backT(i+1);

            col[j] = diagPrin[N + i-j] = diagSec[i+j] = false;
        }
    }
}