八皇后问题

问题的类别是回溯(backtrace).

回溯通常用递归实现。回溯中注意边界问题,避免越界。同时,剪枝可以减少case数量。

具体看leetcode这里

// .h
#ifndef AGORITHM_NQUEENS_H
#define AGORITHM_NQUEENS_H

#include <string>
#include <vector>

class NQueens {
public:
    static void test_nqueens();

    static std::vector<std::vector<std::string>> solveNQueens(int n);

private:
    static void visit(std::vector<std::vector<int>> &vis, std::vector<std::vector<int>> &mat, int r, int c,
                      std::vector<std::vector<std::string>> &ans);


    static void generateAnswer(const std::vector<std::vector<int>> &mat, std::vector<std::vector<std::string>> &ans);
};
#endif //AGORITHM_NQUEENS_H
// .cpp
#include <sstream>
#include <iostream>
#include "NQueens.h"

using namespace std;

std::vector<std::vector<std::string>> NQueens::solveNQueens(int n) {
    vector<vector<int>> vis(4, vector<int>(n * 2, 0));
    std::vector<std::vector<std::string>> ans;
    vector<vector<int>> mat(n, vector<int>(n, 0));

    for (int i = 0; i < n; i++) {
        visit(vis, mat, 0, i, ans);
    }

//    int nAnswer = ans.size();

    return ans;
}

void NQueens::visit(std::vector<std::vector<int>> &vis, std::vector<std::vector<int>> &mat, int r, int c,
                    std::vector<std::vector<std::string>> &ans) {
    const int N = mat.size();

    if (!vis[0][r] && !vis[1][c] && !vis[2][r + c] && !vis[3][(r - c + 2 * N) % (2 * N)]) {
        // mark
        vis[0][r] = 1;  // kth row
        vis[1][c] = 1;  // mth column
        vis[2][r + c] = 1;
        vis[3][(r - c + 2 * N) % (2 * N)] = 1;
        mat[r][c] = 1;
        if (r == N - 1) {
            generateAnswer(mat, ans);
        } else {
            for (int j = 0; j < N; j++) {
                visit(vis, mat, r + 1, j, ans);
            }
        }

        vis[0][r] = 0;
        vis[1][c] = 0;
        vis[2][r + c] = 0;
        vis[3][(r - c + 2 * N) % (2 * N)] = 0;
        mat[r][c] = 0;
    }
}

void NQueens::generateAnswer(const std::vector<std::vector<int>> &mat, std::vector<std::vector<std::string>> &ans) {
    const int N = mat.size();
    std::vector<std::string> vec;
    for (int i = 0; i < N; i++) {
        std::string s;
        for (int j = 0; j < N; j++) {
            if (mat[i][j]) {
                s += "Q";
            } else {
                s += ".";
            }
        }
        vec.emplace_back(s);
    }
    ans.emplace_back(vec);
}

void NQueens::test_nqueens() {
    int n;
    std::cin >> n;
    auto ans = solveNQueens(n);
    std::cout << "nanswer: " << ans.size() / 4 << std::endl;
    for (auto &an : ans) {
        for (const auto &s : an) {
            std::cout << s << "\n";
        }
        std::cout << std::endl;
    }
}

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 问题介绍 摘自百度百科八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟...
    逍遥wqy阅读 1,133评论 0 0
  •   八皇后问题,是一个古老而著名的问题,该问题最早由国际西洋棋棋手马克斯·贝瑟尔(Max Bezzel)于1848...
    山阴少年阅读 897评论 0 1
  • 八皇后问题:在8*8的棋盘上放置8个皇后,保证任意两个皇后之间不能相互攻击。(即没有两个皇后是在同一行、同一类、或...
    五秋木阅读 784评论 0 0
  • 目录[TOC] 问题分析: 相信八皇后规则的问题,大家都很熟悉,接下来是如何分析回溯法的应用。回溯法与图里面的深度...
    maskwang520阅读 3,213评论 0 5
  • 八皇后问题问题描述:八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。该问题是国际西洋棋棋手马克斯·贝瑟尔...
    药药耀耀阅读 2,110评论 0 0