JB의 이모저모

[Programmers Lv.2] 이진 변환 반복하기 (Python) 본문

알고리즘/프로그래머스

[Programmers Lv.2] 이진 변환 반복하기 (Python)

J B 2023. 9. 10. 20:50

https://school.programmers.co.kr/learn/courses/30/lessons/70129#

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

이진 변환 반복하기

문제 설명

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

입출력 예

sresult

`"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 "01110" 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" 1 1 "1"
  • 4번의 이진 변환을 하는 동안 1개의 0을 제거했으므로, `[4,1]`을 return 해야 합니다.

 

⭕ CODE

def solution(s):
    answer = []
    # 이진 변환 횟수 
    cnt = 0
    # 제거된 모든 0 의 개수
    zero_cnt = 0
    s_list = list(s)
    # s가 1이 아닌동안
    while s_list != ['1']:
        # 새로운 s는 결국 1의 개수이므로
        new = bin(s_list.count('1'))
        # 제거된 0의 개수 더해주기
        zero_cnt += s_list.count('0')
        # 이진 변환 1번 더해주기
        cnt += 1
        # 새로운 s_list는 0b를 제외한 숫자만
        s_list = list(new)[2::]
        
    answer.append(cnt)
    answer.append(zero_cnt)
    return answer

✏️ Comment

문제에서 요구하는 내용이 0을 다 빼고 나머지의 길이를 다시 이진수로 변환해서 원래 s_list에서 1의 개수가 길이로 생각하여 바로 1의 개수를 bin(이진수처리) 해주었다 bin의 결과는 0b1100 이런 식으로 나오기에 새로운 리스트는 0b를 제외하기 위해 list(new)[2::]을 해주었다.