2048. 下一个更大的数值平衡数

214 阅读2分钟

这是我参与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 的最小数值平衡数。

解题思路

  1. 先计算当前n有多少位,设为k,如果n比拥有k个数位的数值平衡数都要大,那么我们就求k+1位的最小数值平衡数,
  2. 使用回溯法,求出拥有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();
            }
        }
    }

};