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

Jhanoo·2024년 10월 2일

알고리즘 스터디

목록 보기
46/80

[level 2] 이진 변환 반복하기 - 70129

문제 링크

성능 요약

메모리: 84.7 MB, 시간: 7.93 ms

구분

코딩테스트 연습 > 월간 코드 챌린지 시즌1

채점결과

정확성: 100.0
합계: 100.0 / 100.0

제출 일자

2024년 10월 02일 10:23:55

문제 설명

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 제거 후 길이 이진 변환 결과
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 해야 합니다.

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges


풀이

  • s의 길이는 최대 150,000이므로 첫 변환은 문자열에서 0의 개수와 1의 개수를 구한다.
  • 첫 연산 이후부터는 int 범위에 속하는 수이므로 비트연산을 통해 1의 개수와 0의 개수를 구한다.

코드

class Solution {
    public int[] solution(String s) {
		int[] answer = { 0, 0 };

		int num = 0;
		for (int i = 0; i < s.length(); i++) {
			if (s.charAt(i) == '1')
				num++;
		}

		answer[0]++; // 횟수
		answer[1] += s.length() - num; // 0의 개수

		// 150,000 부터는 int 범위
		while (num > 1) {
			int[] changed = change(num);
			answer[0]++;
			answer[1] += changed[0];
			num = changed[1];
		}

		return answer;
	}

	public static int[] change(int num) {
		int one = 0;
		int zero = 0;

		while (num > 0) {

			// 마지막 비트가
			if ((num & 1) == 1) // 1이면
				one++;
			else // 0이면
				zero++;

			num >>= 1; // >> 비트시프트 연산 모든 비트를 한 칸씩 오른쪽으로 이동
		}

		return new int[] { zero, one };
	}
}
profile
어떻게든 해내는 사람

0개의 댓글