受伤的皇后——21年模拟赛

题目链接：用户登录https://www.lanqiao.cn/problems/552/learning/?page=3&first_category_id=1&sort=students_count

题目描述

有一个 n×n 的国际象棋棋盘（n 行 n 列的方格图），请在棋盘中摆放 n 个受伤的国际象棋皇后，要求：

1. 任何两个皇后不在同一行。
2. 任何两个皇后不在同一列。
3. 如果两个皇后在同一条 45 度角的斜线上，这两个皇后之间行号的差值至少为 3 。

请问一共有多少种摆放方案。

输入描述

输入的第一行包含一个整数 n。

输出描述

输出一个整数，表示答案。

输入输出样例

示例 

输入

4

输出

2

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

 

JAVA解法 ：

import java.util.Arrays;
import java.util.Scanner;

public class n皇后2 {
    static int n ;
    static int[] a;
    static int count=0 ;
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt()+1;
        a = new int[n];
        dfs(1);
        System.out.println(count);
    }

    private static void dfs(int row) {//第row皇后放何处
        if (row==n){
            count++;
            //System.out.println(Arrays.toString(a));
        }
        for (int i = 1; i <= n-1; i++) {//生成列
            if (check(row,i)){
                a[row]=i;
                dfs(row+1);
                a[row]=0;
            }
        }
    }

    private static boolean check(int row, int col) {
        for (int j = 1; j <= row; j++) {//在已经放过皇后的行进行判断
            if (a[j]==col) return false; //j行a[j]列已有皇后
            if (j+a[j] == row+col) return false;//这个位置的反对角线已有皇后
            if (j-a[j] == row-col) return false;//这个位置的正对角线已有皇后
        }
        return true;
    }

}

C++解法：

#include <iostream>
using namespace std; 
int a[10]={0};//用来存储第i行存储在第几列 
int count,n; 
bool valid(int row,int y)//判断row行的y列是否可用 
{ for(int i=1;i<row;i++)//判断前row-1行中有无与y冲突的列 
    { 
    if(a[i]==y) return false;//前row-1行中有行存在第y列不可用 
    if((a[i]+i==row+y)&&(row-i)<3) return false;//前row-1行中有列存在和第y列在反对角线上不可用 
    if((row-i==y-a[i])&&(row-i)<3) return false;//前row-1行中有列存在和第y列在正对角线上不可用
     } 
 return true; 
 } 
 void dfs(int row) 
 { 
    if(row==n+1)//前row行均以排完，方案数加一
      { 
        count++; 
        return; 
      } 
    for(int i=1;i<=n;i++)//依次判断1到n列有无可用
   { 
     if(valid(row,i))//row行i列可用
      { 
      a[row]=i;//row行存在第i列
     dfs(row+1);//处理下一行
      a[row]=0;//重置便与尝试新方案
       } 
    } 
  } 
int main() 
{ 
  cin>>n; 
  dfs(1); 
  cout<<count; 
  return 0;//给楼里一位兄弟的代码加了些个人认为的注释，便于各位理解
}

python解法：

import os
import sys
import math
# 请在此输入您的代码
# 使用dfs进行深度搜索，其中会有多种重复
# 用dfs深搜的结果 / 行数的阶乘可得正确答案
def isleagal(x,y):
  if 0 <= x < n and 0 <= y < n :
    return True
  return False

def isrowcol(i,j):  # 行列没有返回True
  for k in range(n):
    if v[i][k] ==1 or v[k][j] == 1:
      return False
  return True

def isxie(i,j): # 45度对角线没有返回True
  dx = [1,1,-1,-1]
  dy = [1,-1,-1,1]
  for k in range(4):
    for m in range(1,3):
      nx = i+dx[k]*m
      ny = j+dy[k]*m
      if isleagal(nx,ny) and v[nx][ny] ==1:
        return False
  return True



def dfs(x):

  if x == n:  # 当放的皇后的数量x == n时结束
    ans[0] +=1
    return

  for i in range(n):
    for j in range(n):
      if v[i][j] == 0 and isrowcol(i,j) and isxie(i,j):
        v[i][j]=1
        dfs(x+1)
        v[i][j]=0





ans =[0]
n = int(input())
num = 1
for i in range(1,n+1):
    num *= i
v = []
for i in range(n):
  v.append([0] * n)
dfs(0)
print(ans[0]//num )