프로그래머스_이진 변환 반복하기(JAVA)

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

[문제 풀이]

s가 1이 될때까지 while문을 돌려 이진 변환을 반복하여 풀었다.

처음 문제를 푼 방식으로 하면 테스트케이스는 잘 해결되는데 최종 테스트에서 런타임 에러가 발생하였다.

z_num 변수는 제거된 모든 0의 개수, count 변수는 총 이진 변환 횟수를 count할 변수이다.

런타임 에러는 Integer.bitCount(Integer.parseInt(s,2)) 이 과정에서 발생한 것 같다.

string s를 2진수로 간주하고 이를 다시 10진수로 변환한 Integer.parseInt(s,2)에서 Integer.bitCount()를 이용하여 1의 개수를 count하였는데 이 부분에서 시간을 많이 요했다.

그래서 이를 이용하지 않고 for문을 이용해 1과 0의 개수를 하나하나 count하니 잘 해결되었다.

z_num은 총 제거한 0의 개수이므로 0개수를 각 회차마다 누적시켜 더해주고, s는 기존 s에서 0을 제거하고 난 후 1의 개수를 이진 변환한 것이므로 for문에서 count한 1의 개수를 Integer.toBinaryString()을 이용하여 2진 string으로 변환시켜주면 된다.

s가 "1"이 되면 반복문을 빠져나오고 최종 반환할 answer 배열의 0번째 원소에는 총 이진 변환 횟수인 count를, 1번째 원소에는 총 제거한 0의 개수인 z_num을 넣어주고 반환하면 된다.

 

 

[런타임 에러]

import java.util.*;
class Solution {
    public int[] solution(String s) {
        int[] answer = new int[2];
        int z_num =0, count=0;
        
        while(!s.equals("1")){
            z_num += s.length()-Integer.bitCount(Integer.parseInt(s,2));
            s = Integer.toBinaryString(Integer.bitCount(Integer.parseInt(s,2)));
            count++;
        }
        
        answer[0]=count;
        answer[1]=z_num;
        return answer;
    }
}

 

[최종 코드]

import java.util.*;
class Solution {
    public int[] solution(String s) {
        int[] answer = new int[2];
        int z_num =0, count=0, o_num=0;
        
        while(!s.equals("1")){
            o_num=0;
            for(int i=0;i<s.length();i++){
                if(s.charAt(i)=='1')
                    o_num++;
                else
                    z_num++;
           }
            s = Integer.toBinaryString(o_num);
            count++;
        }
        
        answer[0]=count;
        answer[1]=z_num;
        return answer;
    }
}