Cod sursa(job #2253540)

Utilizator dnprxDan Pracsiu dnprx Data 4 octombrie 2018 09:29:13
Problema Problema Damelor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.18 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("damesah.in");
ofstream fout("damesah.out");
int st[25], n;
int viz[25], dp[50], ds[50], C;

/**
 O dama de la pozitia (i,j) se afla pe
  - a (n-j+i)-a diagonala principala
  - a (i+j-1)-a diagonala secundara

 Pentru orice i=1..2n-1
 dp[i] = 1 daca a i-a diagonala principala e ocupata
 ds[i] = 1 daca a i-a diagonala secundara e ocupata

*/
void Afis()
{
    C++;
    if(C == 1)
    {
        for(int i = 1; i <= n; i++)
            fout << st[i] << " ";
        fout << "\n";
    }
}

inline int Valid(int i, int j)
{
    if(viz[j] == 1)return 0;
    if(dp[n - j + i] == 1)return 0;
    if(ds[i + j - 1] == 1)return 0;
    return 1;
}

void Back(int top)
{
    if(top == n + 1)Afis();
    else for(int j = 1; j <= n; j++)
        if(Valid(top , j))
        {
            st[top] = j;
            viz[j] = 1;
            dp[n - j + top] = 1;
            ds[top + j - 1] = 1;
            Back(top + 1);
            dp[n - j + top] = 0;
            ds[top + j - 1] = 0;
            viz[j] = 0;
        }
}

int main()
{
    fin >> n;
    Back(1);
    fout << C << "\n";
    return 0;
}