Python/Coding Test

[Coding Test] 이진 변환 반복하기

gangee 2023. 7. 7. 02:45
728x90
반응형

프로그래머스 코딩테스트 2단계 : 이진 변환 반복하기

문제 설명

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의 길이가 아닌 s를 2진법으로 변환
  • 시간 초과 발생

정답 풀이

  • 정답 리스트, 이진변환 실행횟수를 저장할 cnt, 문자열의 0의 개수를 저장할 zero_cnt 정의
  • while 반복문을 통해 반복문 안의 코드를 반복하고, s가 '1' 일 때 반복문 종료
  • count 함수를 통해 s 문자열의 0의 개수를 zero_cnt에 더하기
  • replace 함수를 통해 0을 삭제하고 s에 다시 저장
  • len 함수를 통해 문자열의 길이를 구하고 bin 함수를 통해 2진수로 변환
  • bin() : 이진 변환 함수, 실행하면 숫자 앞에 ob가 붙음, 그래서 [2:]을 통해 리스트 자름
  • 위에 3개를 한번 반복하고 나면 cnt에 1을 더해줌
  • 반복문이 끝나면 answer 리스트에 [cnt, zero_cnt]를 저장하고 반환

정답 코드

def solution(s):
    answer = []
    cnt = 0
    zero_cnt = 0

    while True:

        zero_cnt = zero_cnt + s.count('0')
        s = s.replace('0', '')
        s = bin(len(s))[2:]
        cnt += 1

        if s == '1':
            break

    answer = [cnt, zero_cnt]
    return answer
* 이 문제는 프로그래머스 코딩테스트 연습 2단계 문제입니다.
728x90
반응형