这是我参与11月更文挑战的第10天,活动详情查看:2021最后一次更文挑战
2048. 下一个更大的数值平衡数
如果整数 x 满足:对于每个数位 d ,这个数位 恰好 在 x 中出现 d 次。那么整数 x 就是一个 数值平衡数 。
给你一个整数 n ,请你返回 严格大于 n 的 最小数值平衡数 。
示例 1:
输入:n = 1
输出:22
解释:
22 是一个数值平衡数,因为:
- 数字 2 出现 2 次
这也是严格大于 1 的最小数值平衡数。
示例 2:
输入:n = 1000
输出:1333
解释:
1333 是一个数值平衡数,因为:
- 数字 1 出现 1 次。
- 数字 3 出现 3 次。
这也是严格大于 1000 的最小数值平衡数。
注意,1022 不能作为本输入的答案,因为数字 0 的出现次数超过了 0 。
示例 3:
输入:n = 3000
输出:3133
解释:
3133 是一个数值平衡数,因为:
- 数字 1 出现 1 次。
- 数字 3 出现 3 次。
这也是严格大于 3000 的最小数值平衡数。
解题思路
- 先计算当前n有多少位,设为k,如果n比拥有k个数位的数值平衡数都要大,那么我们就求k+1位的最小数值平衡数,
- 使用回溯法,求出拥有k个数位的所有数值平衡数,因为我们只能从1-9中取数字组成数值平衡数,并且每个数位的出现次数必须与数位值相同,我们尝试组成k个数位的所有可能性,并且使用全排列将所有的可能排列遍历一遍,找出严格大于 n 的 最小数值平衡数 。
代码
class Solution {
public:
int min=0x3f3f3f3f,tar;
int nextBeautifulNumber(int n) {
this->tar=n;
if(n==0) return 1;
dfs(vector<int>{},0, 1 + (int)log10(n));
dfs(vector<int>{},0, 2 + (int)log10(n));
return min;
}
void dfs(vector<int> arr,int cur,int len){
if (len==0){
do
{
int sum=0;
for (int i = 0; i < arr.size(); ++i) {
sum=sum*10+arr[i];
}
if (sum>tar&&sum<min)
min=sum;
}while (next_permutation(arr.begin(),arr.end()));
return;
}
for (int i = cur; len>=i ; ++i) {
for (int j = 0; j < i; ++j) {
arr.push_back(i);
}
dfs(arr,i+1,len-i);
for (int j = 0; j < i; ++j) {
arr.pop_back();
}
}
}
};