알고리즘/프로그래머스

[프로그래머스 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 이하의 자연수입니다.

 분석

 맨 처음 나는 이진수로 바꾸고 규칙을 찾아 바꾼 이진수 내에서 조건에 맞는 다음 큰 숫자를 찾으려고 했다. 예를 들어 110(6)인 경우, 1001(9)로 바꾼다 든가.. 그러나 해보면 알지만 조건들이 매우 복잡하다. 그래서 방향을 바꿔 10진수로 처리하기로 했다. 

 예를들어 한 숫자가 주어지면, 그 수의 2진수에서 1의 개수를 세고, 그 수보다 큰 숫자들을 하나씩 2진수에서 1의 개수를 세서 같은 경우를 찾아 처리했다.

 

 구현

using namespace std;

// 1의 개수를 반환.
int make_binary(const int n)
{
	int cnt = 0;
	int tmp = n;
	while (tmp)
	{
		if (tmp % 2 == 1) cnt++;
		tmp /= 2;
	}
	return cnt;
}

int solution(int n)
{
	int answer = n;
	int cnt_orgin = make_binary(answer), cnt_compare = 0;
	
	while (true)
	{
		cnt_compare = make_binary(++answer);
		if (cnt_orgin == cnt_compare) break;
	}
	
	return answer;
}

int main()
{
	cout << solution(6);
	return 0;
}