c++ n皇后问题

Posted 2022-06-30 lja001162

题目描述

会下国际象棋的人都很清楚：皇后可以在横、竖、斜线上不限步数地吃掉其他棋子。如何将8个皇后放在棋盘上（有8 * 8个方格），使它们谁也不能被吃掉！这就是著名的八皇后问题。

输入

一个整数n（ 1 < = n < = 10 )

输出

每行输出对应一种方案，每种方案顺序输出每一行皇后所在的列号，相邻两数之间用空格隔开，按字典序输出。如果不存在对应的方案，输出-1。

样例输入

4

样例输出

2 4 1 3
3 1 4 2

Source Code

#include <iostream>
#include <cmath>
#include <string.h>
#include <stdio.h>
using namespace std;
int n,m,a[100],s[100],sum,ans;
bool isPrime(int);
void dfs(int x,int y)

    if(x == m)//如果挑选出来3组数据,就求和,小于3个数,就继续挑选
    
        sum = 0;
        for(int i = 1;i <= x;i ++)
        
            printf("a[%d] = %d\n",i,a[i]);
            sum += s[a[i]];//对挑出来的三个数t求和
        
        cout << "------------" << endl;
        if(isPrime(sum) == 1)
            ans ++;//如果是和是素数,就+1
        return ;
    
    y ++;
    for(int i = y;i <= n;i ++)//从4个数中挑选三个数字
    
        a[x + 1] = i;//改变下标形成新的挑选数
        dfs(x + 1,i);//当x + 1等于3 就开始求和  如果小于3就继续挑选
    

bool isPrime(int r)//判断是否是素数

    for(int i = 2;i <= sqrt(r);i ++)
        if(r % i == 0)
            return 0;
    return 1;

int main()

    cin >> n >> m;
    for(int i = 1;i <= n;i ++)
        cin >> s[i];
    dfs(0,0);
    cout << ans << endl;
    return 0;