[프로그래머스] 이진 변환 반복하기

알고리즘 · 2021. 1. 27.

문제 설명

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'이 최소 하나 이상 포함되어 있습니다.

입/출력

입/출력


풀이

def solution(s):
    from collections import Counter
    zero,cnt = 0,0
    
    while s != '1':
        cnt += 1
        zero += Counter(s)['0']
        s = str(bin(int(Counter(s)['1']))).replace("0b","")
        #s = str(bin(int(Counter(s)['1'])))[2:]
        
    
    return [cnt,zero]
function solution(s) {
    let zero = 0
    let cnt = 0
    
    while(s !== '1'){
        const temp = [...s].reduce((total,string)=>{
          if(string === '0'){
              zero += 1
              return total
          }  
            return total +1
        },0)
        s = temp.toString(2)
        cnt += 1
    }
    
    return [cnt,zero];
}

결과


코드 풀이

자바스크립트를 이용해서 문제 푸는게 아직 익숙하지 않아서 함수를 생각해내지도 못했다 . spread array 와 reduce 를 기억해야겠다.

  1. 파이썬은 Counter 라이브러리를 이용해서 풀 수 있다.
  2. 2진 변환을 한 것이 1이 될 때 까지 while문을 계속한다.
  3. Counter함수에 지워진 0의 갯수를 zero에 누적해서 더해준다
  4. s에 bin함수를 이용한 이진 변환을 이용한 수를 담는다 . 여기서 replace("0b","")를 해도 정답 [2:] 배열 슬라이싱도 정답이다.
  5. while문이 끝나면 배열에 담은 cnt 와 zero를 retrun 한다.