본문 바로가기

프로그래머스(LV2)

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

1. 문제설명

  • 0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다.
    •  
    • x의 모든 0을 제거합니다.
    • 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'이 최소 하나 이상 포함되어 있습니다.
  • 입출력 예
    "110010101001" [3,8]
    "01110" [3,3]
    "1111111" [4,1]

 

 

2. 문제 해결

def solution(s):
    answer = []
    cnt = 0     
    zeroCnt = 0
    
    #"1"이 남을 때까지 반복
    while True:
        if s == "1":
            break;
            
        zeroCnt += s.count("0")  #문자열의 0의 개수 세기
        s = s.replace("0",'')    #0을 공백으로 변환
        s = bin(len(s))[2:]      #2진수로 변환
        
        cnt +=1
        
    answer = [cnt, zeroCnt]    
    return answer

접근방식

  • 문자열 s가 들어왔을 때 0의 갯수를 세고, 0을 제거한다.
  • 제거된 문자열 s의 길이를 2진수로 변환하고 위의 작업을 "1"이 남을 때까지 반복한다.
  • s.count("0")로 문자열 s의 0의 갯수를 세고 zeroCnt에 저장한다.
  • s.replace("0", '')를 통해 문자열 s의 0을 공백으로 치환한다. (예시 : 1010 -> 11)
  • bin( ) 함수는 이진수로 변환해주는 역할을 한다. 
  • bin(len(s))은 문자열의 s의 길이를 이진 문자열로 변환하게 한다.
  • bin(len(s))[2:]은 이진 문자열을 알려주는 "0b"을 생략하게 해준다.

 

 

 

3. 다른 사람 풀이

def solution(s):
    a, b = 0, 0
    while s != '1':
        a += 1
        num = s.count('1')
        b += len(s) - num
        s = bin(num)[2:]
    return [a, b]

해석

  • a = 몇회차에 '1'로 끝났는지
  • b = s의 총 길이에서 숫자 1의 갯수만큼 뺌  = 0의 갯수

 

내 코드와 다른 사람의 코드가 다른점

  • while 조건문을 != '1'로 하여 불필요한 if문을 쓰지 않았다.
  • 문자열 s의 '1'의 갯수를 세면 그 변수를 이용하여 바로 이진 문자열로 변환이 가능하다.
  • 불필요하게 replace("0", '')으로 공백으로 바꿀 이유가 없다.
  • 이분의 코드가 더 깔끔하고, 불필요한 연산 없이 깔끔하다.