Programmers - 이진 변환 반복하기(Python)

문제 설명

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"66
2"110"12
3"10"11
  • 3번의 이진 변환을 하는 동안 8개의 0을 제거했으므로, [3,8]을 return 해야 합니다.

입출력 예 #2

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

입출력 예 #3

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

✍ 코드

def solution(s):
    answer = []

    # (1)
    cnt_binary = 0

    # (2)
    cnt_zero = 0

    while 1:
        # (3)
        if s == "1":
            break

        # (4)
        cnt_binary += 1

        # (5)
        cnt_zero += s.count("0")

        # (6)
        s = s.replace("0", "")

        # (7)
        len_s = len(s)

        # (8)
        s = bin(len_s)[2:]

    # (9)
    answer.append(cnt_binary)
    answer.append(cnt_zero)

    return answer


s = "110010101001"

print(solution(s)) # 결과 예 : [3,8]

(1) : 이진 변환 횟수를 파악하기 위한 변수 cnt_binary
(2) : 각 회차마다 s에 존재하는 0의 갯수를 파악하기 위한 변수 cnt_zero
(3) : while문 종료 조건
(4) : while문이 종료가 안된다면 cnt 1 증가
(5) : s에 존재하는 "0"의 갯수를 파악한 후, 이를 cnt_zero에 더해줌
(6) : 각 회차마다 s에 존재하는 0을 제거하기 위해 replace() 메소드 활용
(7) : s에 존재하는 1의 갯수가 len(s)와 동일하며, 이를 len_s 변수에 할당
(8) : len_s를 2진수로 변환한 값을 다시 s에 대입
(9) : while문이 종료되고 최종적으로 구해진 cnt_binary, cnt_zero를 answer에 할당

📝 결과

😃 느낀점

  • count, replace, bin 등의 메소드를 과거에 공부했던 적이 있어서, 이번 문제는 상대적으로 쉬운 축에 속했던 것 같다.

👍 Irish의 모든 코드 보기

-> Irish Github

좋은 웹페이지 즐겨찾기