코테

[Algorithm] 안전지대- JavaScript

뚜따따 2024. 3. 13. 20:36

프로그래머스 - 안전지대
난이도 : Level 1

 

▶ 풀이

길이를 기준으로 반복문을 돌리며 w * h 의 값이 brown + yellow 의 값인 수를 찾고 그 수 중 테두리를 제외한 높이와 넓이가 노란 타일의 수와 같을 경우 출력한다.

function solution(board) {
  // 위, 아래, 좌, 우, 대각선 칸 확인을 위한 세팅
  const dx = [1, 0, -1, 0, 1, 1, -1, -1];
  const dy = [0, 1, 0, -1, 1, -1, 1, -1];

  // 정사각형이기 때문에 코드 간소화를위해 길이 세팅
  const n = board.length;

  // 초기 안대지대를 모든 칸으로 답을 설정
  let answer = n * n;

  // 지뢰가 있는 곳을 확인해주기 위한 설정
  const ch = [];

  // 이중 for문을 이용해 모든 칸을 확인
  for (let i = 0; i < n; i++) {
    for (let j = 0; j < n; j++) {
      if (board[i][j] === 1) {
        // 지뢰를 발견했다면 ch에 그 칸의 위치 확인
        ch.push([i, j]);
        // 안전지대 -1
        answer--;
      }
    }
  }

  // 모든 지대가 지뢰지역이면 0을 반환
  if (answer === 0) return 0;

  // 위험지역 확인을 위해 지뢰의 위치 확인
  ch.forEach((w) => {
    // 위, 아래, 좌, 우, 대각선 총 8곳의 위치를 확인
    for (let k = 0; k < 8; k++) {
      const nx = w[0] + dx[k];
      const ny = w[1] + dy[k];
      if (nx >= 0 && ny >= 0 && nx < n && ny < n && board[nx][ny] === 0) {
        // 현재 위치를 위험지대로 설정하고
        board[nx][ny] = 1;
        // 안전지대 갯수에서 -1
        answer--;
      }
    }
  });
  return answer;
}