问题描述:
在8×8的国际象棋摆放8个皇后,使其不能互相攻击。
有8个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
洛谷链接:[USACO1.5]八皇后 Checker Challenge - 洛谷
分析:
第一个想法就是8个for循环穷举,但如果使用穷举法需要尝试88=16,777,216种情况。
所以使用回溯法:
回溯算法求解八皇后问题的原则是:有冲突解决冲突,没有冲突往前走,无路可走往回退,走到最后是答案。
重点:需要建立四个标志数组;
line: 存行
row: 存列,row[j] ==1表示第 j列已经有棋子了
left: 存左下角到右上角的对角线,left[i+j] == 1表示这一对角线已经有棋子了(你们可以画图看一下,每根对角线上的点合是固定)
right: 存右下角到左上角的对角线,right[i-j+n] ==1表示这一对角线已经有棋子了,同上,可以画图验证下。
代码+注释:
import java.util.*;
import java.io.*;
public class Main {
static int ans=0;
static int n=0;
static int[] line;
static int[] row;
static int[] left;
static int[] right;
public static void main(String[] args)
{
Scanner scan = new Scanner(System.in);
line = new int[100];//存行
row = new int[100];//存列
left = new int[100];//存左下到右上
right = new int[100];//存右下到左上
n = scan.nextInt();
dfs(1);
System.out.println(ans);
}
public static void dfs(int i)
{
if(i>n) //已经存完棋子,输出
{
print();
return;
}
for(int j=1;j<=n;j++) //i是行,i是列
{
if(row[j]!=1 && left[i+j]!=1 && right[i-j+n]!=1)
{
line[i]=j; //注意这里是j,
row[j]=1;
left[i+j]=1;
right[i-j+n]=1;
dfs(i+1); //继续搜
//回溯
line[i]=0;
row[j]=0;
left[i+j]=0;
right[i-j+n]=0;
}
}
}
public static void print() //输出前三个答案
{
if(ans<3)
{
for(int i=1;i<=n;i++) System.out.print(line[i] + " ");
System.out.println();
}
ans++;
}
}
输入: 6
输出:
2 4 6 1 3 5 3 6 2 5 1 4 4 1 5 2 6 3 4