Cod sursa(job #2309199)

Utilizator SirStevensIonut Morosan SirStevens Data 28 decembrie 2018 16:26:57
Problema Problema Damelor Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <iostream>
#include <vector>
#include <fstream>
#include <cmath>

#define NMAX 14

using namespace std;

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

int n, cnt;
bool col[NMAX], main_diag[NMAX], sec_diag[NMAX];
vector <int> sol;

void printSol(){
    for(int i= 1; i < sol.size(); i++){
        out << sol[i] << " ";
    }
    out << '\n';
}

bool cont(int c, int k){
    if(!col[c] && !main_diag[c - k + n - 1] && !sec_diag[c + k])
            return true;

    return false;
}

void back(int k){
    if(k == n + 1){
       // cout << cnt << " ";
        cnt ++;
        if(cnt == 1){
            printSol();
        }
    }
    else {
        for(int i = 1; i<=n; i++ ){

          //  printSol();
            if(cont(i, k)){
                sol.push_back(i);
                col[i] = main_diag[i - k + n -1] = sec_diag[i + k] = true;
                back(k+1);
                col[i] = main_diag[i - k + n - 1] = sec_diag[i + k] = false;

                sol.pop_back();
            }

        }
    }
}

int main() {
    in >> n;
    sol.push_back(-1);
    back(1);
    out << cnt << " ";
    return 0;
}