P1219 八皇后 Checker Challenge

题目描述:

一个如下的 6×66×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 2 4 6 1 3 52 4 6 1 3 5 来描述，第 𝑖i 个数字表示在第 𝑖i 行的相应位置有一个棋子，如下：

行号 1 2 3 4 5 61 2 3 4 5 6

列号 2 4 6 1 3 52 4 6 1 3 5

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 33 个解。最后一行是解的总个数。

代码:

package lanqiao;

import java.util.*;

public class Main {

    static int arr[][] = new int[14][14];
    static int ans_sum = 0;
    static int N = 0;

    static int ans[] = new int[14]; //存答案
    static int d[] = new int[14];//标记列站位
    static int t[] = new int[28]; //标记右下对角线：x - y差值不变
    static int r[] = new int[28]; //标记左下对角线：x + y相加相同

    public static void dfs(int x)
    {
        if(x == N + 1)
        {
            if(ans_sum < 3){
                for(int i = 1;i <= N;i ++)
                {
                    System.out.print(ans[i] + " ");

                }
                System.out.println();
            }
            ans_sum += 1;
            return;
        }

        for(int y = 1;y <= N;y ++)
        {
            if(d[y] == 0 && t[x-y+N] == 0 && r[x + y] ==0){//是否该位置不能放
//                arr[x][y] ++;
                ans[x] = y;
                d[y] = 1;
                t[x-y+N] = 1;
                r[x+y] = 1;
                dfs(x + 1);
                t[x-y+N]=0;
                r[x+y]=0;
                d[y]=0;
//                arr[x][y] --;
                ans[x] = 0;
            }
        }
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        dfs(1);
        System.out.println(ans_sum);
    }
}