Cod sursa(job #2294982)

Utilizator TooHappyMarchitan Teodor TooHappy Data 2 decembrie 2018 23:32:42
Problema Problema Damelor Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <bits/stdc++.h>
 
using namespace std;
 
ifstream in("damesah.in");
ofstream out("damesah.out");

int n;
vector< int > sol;
vector< vector< int > > sols;

bool valid(int k) {
    for(int i = 1; i < k; ++i) {
        if(sol[i] == sol[k] || (abs(sol[k] - sol[i]) == abs(k - i))) {
            return false;
        }
    }

    return true;
}

void bkt(int k) {
    if(k == n + 1) {
        sols.push_back(sol);
    } else {
        for(int i = 1; i <= n; ++i) {
            sol[k] = i;

            if(valid(k)) {
                bkt(k + 1);
            }
        }
    }
}

int main() {
    ios::sync_with_stdio(false); in.tie(0); out.tie(0);
    
    in >> n;

    sol.resize(n + 1);
    bkt(1);

    for(int i = 1; i <= n; ++i) {
        out << sols[0][i] << " ";
    }
    out << "\n" << (int)sols.size() << "\n";

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