N-Qeen - 시뮬레이션

programmers.co.kr/learn/courses/30/lessons/12952

코딩테스트 연습 - N-Queen

문제 설명

가로, 세로 길이가 n인 정사각형으로된 체스판이 있습니다. 체스판 위의 n개의 퀸이 서로를 공격할 수 없도록 배치하고 싶습니다.

예를 들어서 n이 4인경우 다음과 같이 퀸을 배치하면 n개의 퀸은 서로를 한번에 공격 할 수 없습니다.

 

체스판의 가로 세로의 세로의 길이 n이 매개변수로 주어질 때, n개의 퀸이 조건에 만족 하도록 배치할 수 있는 방법의 수를 return하는 solution함수를 완성해주세요.

제한사항

• 퀸(Queen)은 가로, 세로, 대각선으로 이동할 수 있습니다.
• n은 12이하의 자연수 입니다.

---

입출력 예

n/result

4

2

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

 

#include <string>
#include <vector>

using namespace std;
int dx[]={1,-1,0,0,1,1,-1,-1};
int dy[]={0,0,1,-1,1,-1,1,-1};
int count =0;

void plustarget(vector<vector<int>>&check, int i , int j, int n){
 
    check[i][j]+=1;
    for(int t=0; t<8; t++){
    
        int nx=i+dx[t]; int ny=j+dy[t];
   
        while(nx>=0 && nx<n && ny>=0 && ny<n){
            
            check[nx][ny]+=1;
            nx = nx+dx[t];
            ny = ny+dy[t];
          
         }
        
    }
    
    
}

void minustarget(vector<vector<int>>&check, int i , int j, int n){
    
//    int nx=i; int ny=j;
    check[i][j]-=1;
    for(int t=0; t<8; t++){
    
        int nx=i+dx[t]; int ny=j+dy[t];
   
        while(nx>=0 && nx<n && ny>=0 && ny<n){
          
            check[nx][ny]-=1;
            nx = nx+dx[t];
            ny = ny+dy[t];
         }
        
    }
    
    
}


void find(vector<vector<int>>&check, int n , int lev){
   
    if(lev==n){
        count++;
        return;
    }
    for(int j=0; j<n; j++){
        if(check[lev][j]==0){
            plustarget(check,lev,j,n);
            find(check,n,lev+1);
            minustarget(check,lev,j,n);
        }
    }
    return;
}


int solution(int n) {
    int answer = 0;
    vector<vector<int>>check(n,vector<int>(n,0));
    find(check,n,0);
    answer = count;
    return answer;
}