Cod sursa(job #1991570)

Utilizator MaligMamaliga cu smantana Malig Data 17 iunie 2017 14:08:49
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;
bool col[NMax],diagPrin[2*NMax],diagSec[2*NMax];
int queen[NMax];

void backT(int);

int main() {
    in>>N;

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

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

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

        ++nrSol;
        return;
    }

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

            diagPrin[nr-j+NMax] = diagSec[nr+j] = col[j] = 1;

            queen[nr] = j;
            backT(nr+1);

            diagPrin[nr-j+NMax] = diagSec[nr+j] = col[j] = 0;
        }
    }
}