프론트엔드 개발/Algorithm

프로그래머스 lv.2 이진 변환 반복하기

하이고니 2023. 1. 19. 20:12
문제 설명

0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다.

  1. x의 모든 0을 제거합니다.
  2. x의 길이를 c라고 하면, x를 "c를 2진법으로 표현한 문자열"로 바꿉니다.

예를 들어, x = "0111010"이라면, x에 이진 변환을 가하면 x = "0111010" -> "1111" -> "100" 이 됩니다.

0과 1로 이루어진 문자열 s가 매개변수로 주어집니다. s가 "1"이 될 때까지 계속해서 s에 이진 변환을 가했을 때, 이진 변환의 횟수와 변환 과정에서 제거된 모든 0의 개수를 각각 배열에 담아 return 하도록 solution 함수를 완성해주세요.


 
 
제한사항
  • s의 길이는 1 이상 150,000 이하입니다.
  • s에는 '1'이 최소 하나 이상 포함되어 있습니다.

 

입출력 예
 
s result
"110010101001" [3,8]
"01110" [3,3]
"1111111" [4,1]

 
 
입출력 예 설명

 

입출력 예 #1

  • "110010101001"이 "1"이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차 이진 변환 이전 제거할 0의 개수 0 제거 후 길이 이진 변환 결과
1 "110010101001" 6 6 "110"
2 "110" 1 2 "10"
3 "10" 1 1 "1"
  • 3번의 이진 변환을 하는 동안 8개의 0을 제거했으므로, [3,8]을 return 해야 합니다.

 

 

입출력 예 #2

  • "01110"이 "1"이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차 이진 변환 이전 제거할 0의 개수 0 제거 후 길이 이진 변환 결과
1 "011110" 2 3 "11"
2 "11" 0 2 "10"
3 "10" 1 1 "1"
  • 3번의 이진 변환을 하는 동안 3개의 0을 제거했으므로, [3,3]을 return 해야 합니다.

 

 

입출력 예 #3

  • "1111111"이 "1"이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차 이진 변환 이전 제거할 0의 개수 0 제거 후 길이 이진 변환 결과
1 "1111111" 0 7 "111"
2 "111" 0 3 "11"
3 "11" 0 2 "10"
4 "10" 0 1 "1"
  • 4번의 이진 변환을 하는 동안 1개의 0을 제거했으므로, [4,1]을 return 해야 합니다.

 

 

나의 코드

function solution(s) {
  let newStr = '';
  let count = 0;
  let changed = 0;

// s === '1' 일 때까지..
// s 에 0이 없는지 체크
// 있으면
// 순회하면서 1은 newStr에 추가
// 순회하면서 0은 삭제하고, 삭제할 때마다 count++

  while (s !== "1") {
      for(let i = 0; i < s.length; i++) {
        if(s[i] === '1') {
          newStr += s[i];
        } else {
          count++
        }
      }
    s = newStr.length.toString(2);
    newStr = '';
    changed++;
  }
  return [changed, count];
}

 

 

깔끔한 코드

function solution(s) {
    var answer = [0,0];
    while(s.length > 1) {
        answer[0]++;
        answer[1] += (s.match(/0/g)||[]).length;
        s = s.replace(/0/g, '').length.toString(2);
    }
    return answer;
}

 

.match는 괄호 안의 데이터와 같은 것이 있으면 배열에 차곡 차곡 쌓아주는 메서드다.

예를 들어, s가 "10001" 이라면 s.match(/0/g) 의 리턴값은 [0, 0, 0] 이다.

그러므로 answer[1]에는 제거한 0의 개수를 넣을 수 있는 것.