原题链接:869. 重新排序得到 2 的幂 - 力扣(LeetCode) (leetcode-cn.com)
题目描述:
给定正整数 N ,我们按任何顺序(包括原始顺序)将数字重新排序,注意其前导数字不能为零。
如果我们可以通过上述方式得到 2 的幂,返回 true;否则,返回 false。(难度:中等)
方法一:搜索回溯 + 位运算
将 nn 的十进制表示视作一个字符数组,我们可以枚举该数组的所有排列,对每个不含前导零的排列判断其对应的整数是否为 22 的幂。
这可以拆分成两个子问题:
枚举可能包含重复字符的数组的全排列,读者可参考「47. 全排列 II」的官方题解;
判断一个整数是否为 22 的幂,读者可参考「231. 2 的幂」的官方题解。
C++代码:
class Solution {
vector<int> vis;
bool isPowerOfTwo(int n) {
return (n & (n - 1)) == 0;
}
bool backtrack(string &nums, int idx, int num) {
if (idx == nums.length()) {
return isPowerOfTwo(num);
}
for (int i = 0; i < nums.length(); ++i) {
// 不能有前导零
if ((num == 0 && nums[i] == '0') || vis[i] || (i > 0 && !vis[i - 1] && nums[i] == nums[i - 1])) {
continue;
}
vis[i] = 1;
if (backtrack(nums, idx + 1, num * 10 + nums[i] - '0')) {
return true;
}
vis[i] = 0;
}
return false;
}
public:
bool reorderedPowerOf2(int n) {
string nums = to_string(n);
sort(nums.begin(), nums.end());
vis.resize(nums.length());
return backtrack(nums, 0, 0);
}
};
方法二:预处理 + 哈希表
C++代码:
string countDigits(int n) {
string cnt(10, 0);
while (n) {
++cnt[n % 10];
n /= 10;
}
return cnt;
}
unordered_set<string> powerOf2Digits;
int init = []() {
for (int n = 1; n <= 1e9; n <<= 1) {
powerOf2Digits.insert(countDigits(n));
}
return 0;
}();
class Solution {
public:
bool reorderedPowerOf2(int n) {
return powerOf2Digits.count(countDigits(n));
}
};