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

문제 설명

자연수 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	result
78	83
15	23

입출력 예 설명
입출력 예#1
문제 예시와 같습니다.
입출력 예#2
15(1111)의 다음 큰 숫자는 23(10111)입니다.

---

문제 풀이

 

이 문제를 나보다 길게 작성한 사람이 거의 없을 것 같다.

이진수 변환, 변환한 수에서 숫자1을 카운트하는 함수를 직접 작성했었다. 

unsigned long long ChangeToBinary(int n){
    unsigned long long binary = 0;
    
    for(unsigned long long i = 1; n >= 1 ; i *= 10){
		binary += (n % 2) * i;
		n /= 2;
	}
    
    return binary;
}

정수형 변수에 직접 이진수를 넣어주는 작업을 해주는 함수... 주어진 자연수도 백만까지라 unsigned long long 까지 사용해주었다.

int NumberCount(unsigned long long n){ 
    unsigned long long num1 = 0;
    string binary_str = to_string(n);
    
    for(int i = 0; i < binary_str.length(); i++){
        if(binary_str[i] == '1') num1++;
    }
    
    return num1;
}

이진수로 바꿔준 수를 다시 스트링으로 변환한 후 1의 개수까지 카운트 해주는 함수도 작성했다.

unsigned long long ChangeToBinary(int n);
int NumberCount(unsigned long long n);

int solution(int n) {
    int answer = n + 1;
    unsigned long long n_binary = ChangeToBinary(n);
    int n_count = NumberCount(n_binary);
    
    while(1){
        unsigned long long answer_binary = ChangeToBinary(answer);
        int answer_count = NumberCount(answer_binary);
        
        if(n_count == answer_count) return answer;
        
        answer++;
    }
    
}

작성한 함수들을 사용해 반복문으로 결과값을 구해줬었다. 

그런데 이렇게 일일이 작성할 필요 없이, 이진수로 바꿔주고 1을 카운트해주는 함수까지 가지고있는 라이브러리가 있었다.

 

bitset 

비트 연산을 도와주고, 정수와 문자열을 비트로 저장할 수 있는 stl이다.

bitset<크기> 변수명(정수/문자열);
bitset<8> b;
bitset<8> b(26);
bitset<8> b("00011010");

위와 같이 선언한 후 사용해주면 된다. (bitset 헤더파일 필요)

배열처럼 []로 접근할 수 도 있고, 비트 연산자도 똑같이 사용 가능하고, 여러 함수들까지 지원해서 여러모로 편리한 것 같다.

 

#include <bitset>

using namespace std;

int solution(int n) {
    int answer = n+1;
    bitset<20> n_bit(n);
    int n_count = n_bit.count();
    
    while (1){
        bitset<20> a_bit(answer);
        int a_count = a_bit.count();
        
        if(n_count == a_count) return answer;
        answer++;
    }
}

bitset을 사용해 작성해 본 코드이다. 

n이 100만 이하의 자연수이기 때문에, 20비트 크기로 생성해주면 딱 맞았다.

기존에 작성한 코드보다 훨씬 간결해져 보기좋다. 

 

2021-06-03 에 작성한 코드 복기

 

 

 

 

 

출처: 프로그래머스 코딩 테스트 연습, 

https://school.programmers.co.kr/learn/challenges