Cod sursa(job #1780392)

Utilizator elffikkVasile Ermicioi elffikk Data 16 octombrie 2016 09:00:38
Problema Problema Damelor Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <iterator>
using namespace std;

bool validate(vector<int> a) {
    vector<bool> b1(a.size()*2), b2(a.size()*2);
    for (int i = 0; i < a.size(); i++) {
        int d1 = a.size()+a[i]-(i+1);
        int d2 = a.size()+(a.size()-a[i]+1)-(i+1);
        if (b1[d1] || b2[d2]) {
            return false;
        }
        b1[d1] = b2[d2] = true;

    }
    return true;
}

int main() {
    ifstream cin("damesah.in");
    ofstream cout("damesah.out");
    int n, m = 0;
    cin>>n;
    vector<int> a, v;
    for (int i = 0; i < n; i++) {
        a.push_back(i+1);
    }
    do {
        if (validate(a)) {
            m++;
            if (v.size()==0) {
                v = a;
            }
        }

    } while (next_permutation(a.begin(), a.end()));
    for (int i = 0; i < v.size(); i++) {
        cout<<v[i]<<" ";
    }
    cout<<"\n"<<m;
}