HDU 2553 经典N皇后问题

最新推荐文章于 2021-01-31 11:15:50 发布

卜 N G

最新推荐文章于 2021-01-31 11:15:50 发布

阅读量162

点赞数

文章标签： dfs 算法 java

本文链接：https://blog.csdn.net/qq_43619527/article/details/105470540

版权

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=2553

Problem Description
在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。

Input
共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

Output
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

Online Judge Online Exercise Online Teaching Online Contests Exercise Author
F.A.Q
Hand In Hand
Online Acmers
Forum | Discuss
Statistical Charts
Problem Archive
Realtime Judge Status
Authors Ranklist

Search
C/C++/Java Exams
ACM Steps
Go to Job
Contest LiveCast
ICPC@China
Best Coder beta
VIP | STD Contests
Virtual Contests
DIY | Web-DIY beta
Recent Contests
Author Ayangli
Mail Mail 0(0)
Control Panel Control Panel
Sign Out Sign Out
N皇后问题

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 51226 Accepted Submission(s): 21292

Input
共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

Output
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

Sample Input
1
8
5
0

Sample Output
1
92
10

很经典的题目，DFS+回溯，代码如下：


import java.io.*;
import java.util.Scanner;
import java.util.StringTokenizer;

public class Main {

    public static int count;

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int times = 1;
        int[] res = new int[16];
        int n = 1;
        while(n != 15) {
            int[][] meza = new int[n][n];
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    meza[i][j] = 0;
                }
            }

            dfs(n, 0, meza);
            res[times++] = count;

            count = 0;
            n++;
        }
        for (int i = 1; i < res.length; i++) {
            System.out.print(res[i] + ",");
        }
    }

    public static void dfs(int n, int now, int[][] meza) {
        if (now == n) {
            count++;
            return;
        }

        for (int i = 0; i < n; i++) {
            if (canSet(now,i,meza)) {
                meza[now][i] = 1;
                dfs(n,now + 1,meza);
                meza[now][i] = 0;
            }
        }

    }

    public static boolean canSet(int m, int i, int[][]p) {
        for (int j = 0; j < p.length; j++) {
            if (p[m][j] == 1) return false;
        }

        //排除列
        for (int j = 0; j < p[0].length; j++) {
            if (p[j][i] == 1) return false;
        }

        int div = m - i;  //用于判断主对角线
        int sum = m + i;  //用于判断副对角线
        for (int k = 0; k < p.length; k++) {
            for (int j = 0; j < p[0].length; j++) {
                if ((k+j) == sum && p[k][j] == 1) return false;
                if ((k-j) == div && p[k][j] == 1) return false;
            }
        }

        return true;
    }

    public static void show(int[][] meza) {
        StringBuilder s = new StringBuilder();

        for (int i = 0; i < meza.length; i++) {
            for (int j = 0; j < meza[0].length; j++) {
                if (meza[i][j] == 1) {
                    s.append(j+1+" ");
                    break;
                }
            }
        }

        System.out.println(s.toString());
    }
}

但是很遗憾，TLE了，算了，打表吧

数据规模没有给出来，我猜也不会太大，n=14的时候我的计算机跑了半分钟

打表代码：

import java.util.Scanner;

public class 打表 {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int[] res = {0,1,0,0,2,10,4,40,92,352,724,2680,14200,73712,365596};
        int n = sc.nextInt();

        while(n != 0) {
            System.out.println(res[n]);
            n = sc.nextInt();
        }

    }
}