[파이썬] 이진 변환 반복하기

programmers.co.kr/learn/courses/30/lessons/70129

코딩테스트 연습 - 이진 변환 반복하기

 

문제 설명

 

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	"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 제거 후 길이이진 변환 결과

회차	이진 변환 이전	제거할 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 해야 합니다.

 

🎄 나의 풀이

 

def solution(s):
    cnt1, cnt2=0, 0		# 제거된 모든 0의 개수, 이진 변환의 횟수
    
    while s != '1':
        cnt1+=s.count('0')
        s=[i for i in s if i != '0']        # 0 제거
        length=len(s)                       # 0 제거 후 길이
        s=bin(length)[2:]              # 이진 변환 결과
        cnt2+=1						# 이진 변환의 횟수

    return [cnt2, cnt1]

 

s에서 실제로 0을 제거하는 대신 0을 제외한 수들을 리스트로 저장했다.

bin함수를 이용해서 쉽게 이진수로 변환할 수 있다. bin함수는 앞에 0b가 붙기 때문에 슬라이싱을 해줬다.

 

🎋 다른사람 풀이

 

def solution(s):
    a, b = 0, 0 	# 이진 변환의 횟수, 제거된 모든 0의 개수
    while s != '1':
        a += 1
        num = s.count('1')
        b += len(s) - num		# s의 길이-1의 개수=0의 개수
        s = bin(num)[2:]		# 이진 변환
    return [a, b]

 

효율적으로 잘 짠 코드.

나랑 다르게 0을 제거하는 것이 없다. s의 길이-1의 개수로 0의 개수를 구했다. 

 

def solution(s):
    from collections import Counter
    cnt, zero = 0, 0
    while s != '1':
        cnt += 1
        counter = Counter(s)
        zero += counter['0']		# dictionary key '0'의 값(개수)
        s = bin(counter['1'])[2:]
    return [cnt, zero]

 

Counter함수로 문자열s에서 각 문자의 개수를 센다. 아래는 예시.

 

from collections import Counter
s='01110'
counter=Counter(s)
print(counter)			# 출력값 Counter({'1': 3, '0': 2})