[프로그래머스] 다음 큰 숫자

📚 문제 설명

자연수 n이 주어졌을 때, n의 다음 큰 숫자는 다음과 같이 정의 합니다.

조건 1. n의 다음 큰 숫자는 n보다 큰 자연수 입니다.
조건 2. n의 다음 큰 숫자와 n은 2진수로 변환했을 때 1의 갯수가 같습니다.
조건 3. n의 다음 큰 숫자는 조건 1, 2를 만족하는 수 중 가장 작은 수 입니다.

예를 들어서 78(1001110)의 다음 큰 숫자는 83(1010011)입니다.

자연수 n이 매개변수로 주어질 때, n의 다음 큰 숫자를 return 하는 solution 함수를 완성해주세요.

제한사항

• n은 1,000,000 이하의 자연수 입니다.

🔎 입출력 예

💻 처음으로 푼 코드

처음에 n과 n의 다음 큰 숫자를 2진수로 바꾼 후 String 배열에 저장한 후 1의 개수를 구했지만 효율성에서 시간 초과가 났다.
그래서 String 배열 대신 toCharArray()를 사용해서 char 배열에 저장한 후 1의 개수를 구하는 방법으로 푸니까 통과할 수 있었다.

class Solution {
    public int solution(int n) {
        
        int answer = n+1;
        int count1 = 0;
        
        char[] arr1 = Integer.toBinaryString(n).toCharArray();
        for(char c : arr1) {
            if(c == '1') count1++;
        }
        
        int count2 = 0;
        while(answer > n) {
            char[] arr2 = Integer.toBinaryString(answer).toCharArray();
            for(char c : arr2) {
                if(c == '1') count2++;
            }
            
            if(count1 == count2) {
                break;
            } else {
                count2 = 0;
                answer++;
            }
        }
        
        return answer;
    }
}

다른 분들이 푼 코드를 보다가 더 간단하고 좋은 방법을 찾게 되었다. 그래서 다시 한번 코드를 수정했다.

💻 최종 수정 코드

Integer.bitCount 메서드를 활용했다!!!

class Solution {
    public int solution(int n) {
        
      int a = Integer.bitCount(n);
      int answer = n+1;
        
      while(true) {   
        if(Integer.bitCount(answer)==a)
        break;
        answer++;
      }
        return answer;
    }
}

📖 Solution

Integer.bitCount( )는 정수를 매개변수로 넣었을때 그 정수의 1 의 개수를 반환한다.

이렇게 간단한 함수를 몰랐다니... 많이 배워간다 👍🏻