[백준 17140번] 이차원 배열과 연산 - 자바스크립트(nodejs)

https://www.acmicpc.net/problem/17140

17140번: 이차원 배열과 연산

const fs = require('fs');
let [input, ...matrix] = fs.readFileSync('dev/stdin').toString().trim().split('\n');
input = input.split(' ');
const [num, location] = [+input.pop(), input.map(Number)];
matrix = matrix.map((v) => v.split(' ').map(Number));

console.log(solution(matrix));

function solution(arr) {
  if (verification(arr)) {
    return 0;
  }
  let count = 1;
  let answer = R_Calculate(arr);

  if (verification(answer)) {
    return count;
  }

  while (count < 100) {
    count++;
    if (answer.length >= answer[0].length) {
      answer = R_Calculate(answer);
    } else {
      answer = C_Calculate(answer);
    }

    if (verification(answer)) {
      return count;
    }
  }

  return -1;
}

function R_Calculate(arr) {
  const result = countNumber(arr);

  const maxLength = Math.max(...result.map((v) => v.length));

  return result.map((v) => {
    while (v.length !== maxLength) {
      v.push(0);
    }
    return v;
  });
}

function C_Calculate(arr) {
  const reverseMatrix = Array.from(Array(arr[0].length), () => Array(arr.length).fill(0));
  for (let i = 0; i < arr.length; i++) {
    for (let j = 0; j < arr[i].length; j++) {
      reverseMatrix[j][i] = arr[i][j];
    }
  }

  const result = countNumber(reverseMatrix);

  const maxLength = Math.max(...result.map((v) => v.length));

  const reverseResultMatrix = Array.from(Array(maxLength), () => Array(result.length).fill(0));

  for (let i = 0; i < result.length; i++) {
    for (let j = 0; j < result[i].length; j++) {
      reverseResultMatrix[j][i] = result[i][j];
    }
  }

  return reverseResultMatrix;
}

function verification(input) {
  if (
    input.length >= location[0] &&
    input[0].length >= location[1] &&
    input[location[0] - 1][location[1] - 1] === num
  ) {
    return true;
  } else {
    return false;
  }
}

function countNumber(arr) {
  const result = arr.map((v) => {
    const obj = {};

    for (let j = 0; j < v.length; j++) {
      if (v[j] === 0) continue;
      !obj[v[j]] ? (obj[v[j]] = 1) : obj[v[j]]++;
    }

    const value = Object.entries(obj)
      .sort((a, b) => {
        if (a[1] !== b[1]) {
          return a[1] - b[1];
        } else {
          return +a[0] - +b[0];
        }
      })
      .flat()
      .map(Number);

    return value;
  });

  return result;
}

골4 구현문제라 그런지 백준, 프로그래머스 구현 문제 푼 것 중에 이 문제가 가장 귀찮았다. 요것 저것 수정하면 계속 케이스가 말썽이었다.

간단한 줄 알았지만 첫 설계의 중요성을 깨달았다.

 

아래부터 함수를 살펴보면

 

countNumber Func

해당 함수는 배열이 입력으로 들어왔을 때 각 배열의 요소들의 개수를 세서 글에 적힌 대로 정렬한 후 다시 배열 형태로 리턴하는 함수이다.

무의미하게 들어간 0은 무시한다.

ex) [[1,2,3,5]] => [1,1,2,1,3,1,5,1]]

 

R_Calculate Func

R 연산을 하는 함수이다. countNumber Func연산 후 마찬가지로 마지막에 0을 넣어줘 행렬의 형태를 유지하고 리턴한다.

 

C_Calculate Func

C 연산을 하는 함수이다. 배열의 행과 열을 바꾼 후 countNumber Func을 한 후 다시 행과 열을 바꿔 마지막에 0을 넣어줘 행렬의 형태를 유지하고 리턴한다.

 

verification Func

A[r][c]에 들어있는 값이 k을 검증하는 함수이다.

입력 배열의 행렬이 주어진 값 (r, c)값 보다 작아야 인덱스를 구할 수 있으며, 배열에서 해당 (r, c) 의 값이 k인지 확인한 후 

true, false를 리턴하는 함수이다.

 

solution Func

 

1. 연산 시도 전에 verification Func이 성립되면 바로 0을 return 한다.

 

2. 3x3이 첫 입력이므로 r연산을 해준다.

 

3. 첫 연산을 끝냈을 때 verification Func으로 검증한다.

 

4. while문을 돌며 count를 1씩 증가시키고, 행이 열보다 크거나 같으면 R연산을, 작으면 C연산을 한다.

 

5. 연산 결과를 verification Func로 검증한다. => true를 리턴하면 count 리턴 후 함수 종료