Cod sursa(job #1343312)

Utilizator AndreiBadeciAndrei Badeci AndreiBadeci Data 15 februarie 2015 11:51:31
Problema Problema Damelor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <fstream>
#include <cmath>
using namespace std;

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

int sol[15],nsol,n,afis[15],uzc[15],uzdp[30],uzds[30];

bool valid(int k)
{
    int i;
    if(uzc[sol[k]]==1)
        return 0;
    if(uzds[k+sol[k]]==1)
        return 0;
    if(uzdp[sol[k]-k+n]==1)
        return 0;
    return 1;
}

void back(int k)
{
    if(k==n+1)
    {
        nsol++;
        if(nsol==1)
            for(int i=1;i<=n;i++)
                afis[i]=sol[i];
        return;
    }
    for(int i=1;i<=n;i++)
    {
        sol[k]=i;
        if(valid(k))
        {
            uzc[i]=1;
            uzdp[i-k+n]=1;
            uzds[i+k]=1;
            back(k+1);
            uzc[i]=0;
            uzdp[i-k+n]=0;
            uzds[i+k]=0;
        }

    }

}

int main()
{
    fin>>n;
    back(1);
    for(int i=1;i<=n;i++)
        fout<<afis[i]<<' ';
    fout<<'\n';
    fout<<nsol;
    return 0;
}