다른 풀이

강의 듣기 전에 혼자 풀어보았는데

function solution(board, moves) {
  let answer = 0,
    stacks = Array.from({ length: board[0].length }, () => []),
    moveStack = [];

  for (let x of board.reverse()) {
    // [3,5,1,3,1]
    for (let i = 0; i < board[0].length; i++) {
      if (x[i] === 0) continue;
      stacks[i].push(x[i]);
    }
  }

  for (let m of moves) {
    m--;
    if (stacks[m].length === 0) continue; // 해당 번호에 인형 다 꺼냈을때

    let top = stacks[m].pop();
    if (moveStack.length === 0) {
      // moveStack 에 아무것도 없을때
      moveStack.push(top);
      continue;
    }

    let movesTop = moveStack.pop();
    if (movesTop === top) answer += 2;
    else {
      moveStack.push(movesTop, top);
    }
  }
  return answer;
}

저는 처음에 board 배열 형태를 pop하기 쉽게 변형한 뒤에 moves 따라 계산했는데

이 코드로 짜면 이중 for 문 때문에 시간 효율이 안좋을까요??

안녕하세요, 인프런 AI 인턴입니다.

님께서 올려주신 자바스크립트 코드는 ‘보드’ 배열의 각 열을 스택으로 변환하여 ‘moves’ 배열에 따라 인형을 뽑아 처리하는 로직을 담고 있습니다. 이중 for문을 이용해 보드를 먼저 스택으로 변형시키고, 그 다음 for문을 이용해 moves에 따라 로직을 처리하는 구조입니다.

시간 복잡도 측면에서 봤을 때 첫 번째 이중 for문은 보드의 크기가 N x N이라고 할 때 O(N^2)의 시간이 소요되고, 두 번째 for문은 moves의 길이를 M이라고 했을 때 최대 O(M) 시간이 소요됩니다.

결과적으로 최악의 시나리오에서 전체 시간 복잡도는 O(N^2 + M)이 됩니다. 보통의 경우 N과 M이 충분히 작다면 시간 효율성에 큰 문제가 없으나, N이나 M이 매우 큰 경우에는 시간 초과가 발생할 수 있는 가능성이 있습니다.

그러나 실제로 이 시간 복잡도가 문제가 될지는 뽑을 수 있는 인형의 최대 개수, 함수 호출 횟수 등이 얼마나 되는지와 보드의 최대 크기에 따라 달라집니다. 코딩 테스트 환경에서는 일반적으로 시간 제한을 문제에서 명시하므로, 이러한 환경에서는 주어진 시간 제한 내에 수행될 수 있는지를 고려하여 코드를 작성해야 합니다.

성능을 개선하고 싶다면, stacks 배열을 만들 때 불필요한 역순으로 원소를 추가하지 않거나, 인형을 뽑는 동작을 최적화하는 등의 방법을 생각해 볼 수 있습니다.

님의 코드가 실제 문제 상황에서 얼마나 효율적으로 작동하는지는 문제의 사이즈와 문제가 요구하는 시간 제한 등을 고려하여 테스트해보시는 것이 좋습니다.

인프런 커뮤니티 질문&답변