八皇后问题 递归搜索

题意：八皇后问题，在同一行，同一列，对角线上不能放皇后，求一共有多少种放的方法。

#include<iostream>
#include<stdio.h>
#include<string>
#include<algorithm>
#include<math.h>
using namespace std;
int r[20] ; //表示
int cnt = 0;
int n = 8; //默认的八行八列的方阵
int ok;
void search(int m)//搜索第M行，认为前m-1行已经搜索完成。
{
    if(m == n)
    {
        cnt++;
        return ;
    }
    else
    {
        for(int i = 0; i < n; i++)//每次从第一列开始搜索
        {
            r[m] = i;
             ok = 1;
            for(int j = 0; j <m;j++)
            {
                if(r[j] == r[m]||m -j == r[j]-r[m]||j-m == r[j]-r[m])                        //同一列，对角线看是否相等。
                   {
                     ok = 0;
                      break;
                   }
            }
            if(ok == 1)
            search(m+1);  //接着搜索下一行说明本行已经找到合适的位置。
        }
    }
    return ;
}
int main()
{
    search(0);      //默认从第0行开始搜索。
    cout<<cnt<<endl;
}