面试题 05.04. 下一个数

下一个数。给定一个正整数，找出与其二进制表达式中1的个数相同且大小最接近的那两个数（一个略大，一个略小）。

示例1:

 输入：num = 2（或者0b10）
 输出：[4, 1] 或者（[0b100, 0b1]）

示例2:

 输入：num = 1
 输出：[2, -1]

提示:

1. num的范围在[1, 2147483647]之间；
2. 如果找不到前一个或者后一个满足条件的正数，那么输出 -1。

package Solutionmst0504;

import java.util.Arrays;

class Solution {
	public int[] findClosedNumbers(int num) {

		if (num == 2147483647) {
			return new int[] { -1, -1 };
		}

		int[] out = new int[2];
		int bitcount = Integer.bitCount(num);

		for (int i = num + 1; i <= 2147483647; i++) {
			if (Integer.bitCount(i) == bitcount) {
				out[0] = i;
				break;
			}
		}

		out[1] = -1;
		for (int i = num - 1; i > 0; i--) {
			if (Integer.bitCount(i) == bitcount) {
				out[1] = i;
				break;
			}
		}
		return out;
	}

	public static void main(String[] args) {

		Solution sol = new Solution();
		int num = 2147483647;

		System.out.println(Arrays.toString(sol.findClosedNumbers(num)));
	}

}