程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> [leetcode]N-Queens

[leetcode]N-Queens

編輯:C++入門知識

class Solution {
    vector<vector<string>> result;
public:
    bool canPlace(int i, int j, vector<string> &tmp, int n){
        if(i == 0) return true;
        for(int k = 0; k < i; k++){
            if(tmp[k][j] == 'Q') return false;
        }
        for(int p = 0; p < i; p++){
            for(int q = 0; q < n; q++){
                if((p-q == i-j || p+q == i+j) && tmp[p][q] == 'Q'){
                    return false;
                }
            }
        }
        return true;
    }

    vector<int> possiblePlaces(int i, vector<string> &tmp, int n){
        vector<int> places;
        
        for(int j = 0; j < n; j++){
            if(canPlace(i,j,tmp, n)){
                places.push_back(j);
            }
        }
        return places;
    }
    
    void dfs(int level, vector<string> &tmp, int n){
        if(level >= n){
            result.push_back(tmp);
            return;
        }
        
        vector<int> places = possiblePlaces(level, tmp, n);
        int j;
        for(int k = 0; k < places.size(); k++){
            j = places[k];
            tmp[level][j] = 'Q';
            dfs(level + 1, tmp, n);
            tmp[level][j] = '.';
        }
        
    }

    vector<vector<string> > solveNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        result. clear();
        vector<string> tmp(n, string(n, '.'));
        dfs(0, tmp, n);
        
        return result;
    }
};

 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved