/*
 * NQueen.cpp
 *
 *  Created on: 2013年12月23日
 *      Author: nerohwang
 */
//形參rowCurrent表示當前所到的行數
#include<iostream>
#include<fstream>
#include<iomanip>
#include<stdlib.h>
using namespace std;
bool Check(int rowCurrent,int *&NQueen);                         //判斷函數
void Print(ofstream &os,int n,int *&NQueen);                                  //打印函數
void Solve(int rowCurrent,int *&NQueen,int n,int &count, ofstream &os);           //N皇后問題處理函數，index一般初值為0

//判斷函數，凡是橫豎有沖突，或是斜線上有沖突，返回FALSE
bool Check(int rowCurrent,int *&NQueen)
{
    int i = 0;
    while(i < rowCurrent)
    {
        if(NQueen[i] == NQueen[rowCurrent] || (abs(NQueen[i]-NQueen[rowCurrent]) == abs(i-rowCurrent)) )
        {
            return false;
        }
        i++;
    }
    return true;
}

//將所有可能出現(xiàn)的結果輸出文本文檔
void Print(ofstream &os,int n,int *&NQueen)
{
    os<<"一次調用\n";
    for (int i = 0;i < n;i++) {
        for(int j = 0 ; j < n; j++)
        {
            os<<(NQueen[i]==j?1:0);
            os<<setw(2);
        }
        os<<"\n";
    }
    os<<"\n";
}

//核心函數。遞歸解決N皇后問題，觸底則進行打印
void Solve(int rowCurrent,int *&NQueen,int n,int &count, ofstream &os)
{
    if(rowCurrent == n)  //當前行數觸底，即完成了一個矩陣，將它輸出
    {
        Print(os,n,NQueen);
        count++;
    }
    for(int i = 0;  i < n; i++)
    {
        NQueen[rowCurrent] = i;                     //row行i列試一試
        if(Check(rowCurrent,NQueen))
        {
            Solve(rowCurrent+1,NQueen,n,count,os);  //移向下一行
        }
    }
}

int main()
{
    int n;           //問題規(guī)模
    int count = 0;   //解的計數
    cout<<"請輸入問題的規(guī)模N"<<endl;
    cin>>n;
    if(n<4)
    {
        cerr<<"問題規(guī)模必須大于4"<<endl;
        return 0;
    }
    int *NQueen = new int[n];
    ofstream os;
    os.open("result.txt");
    Solve(0,NQueen,n,count,os);
    cout<<"問題的解有"<<count<<"種方法"<<endl;
    os.close();
    return 0;
}