【LeetCode】每日一题：下一个更大的数值平衡树——2023.12.10

题目链接：下一个更大的数值平衡树

题干解析：

数值平衡树：若一个数有K位数，则如果存在

则i个i和j个j随意排列组合都是一个k位数值平衡数，再加上k个k就是k位数值平衡树的所有情况；

思路分析：

这个题可以使用打表法（将所有可能的数值平衡树都罗列出来，再根据n进行选取即可，时间复杂度为，其中C为所罗列的数量，空间复杂度为；

class Solution {
public:
    const vector<int> balance {
        1, 22, 122, 212, 221, 333, 1333, 3133, 3313, 3331, 4444,
        14444, 22333, 23233, 23323, 23332, 32233, 32323, 32332,
        33223, 33232, 33322, 41444, 44144, 44414, 44441, 55555,
        122333, 123233, 123323, 123332, 132233, 132323, 132332,
        133223, 133232, 133322, 155555, 212333, 213233, 213323,
        213332, 221333, 223133, 223313, 223331, 224444, 231233,
        231323, 231332, 232133, 232313, 232331, 233123, 233132,
        233213, 233231, 233312, 233321, 242444, 244244, 244424,
        244442, 312233, 312323, 312332, 313223, 313232, 313322,
        321233, 321323, 321332, 322133, 322313, 322331, 323123,
        323132, 323213, 323231, 323312, 323321, 331223, 331232,
        331322, 332123, 332132, 332213, 332231, 332312, 332321,
        333122, 333212, 333221, 422444, 424244, 424424, 424442,
        442244, 442424, 442442, 444224, 444242, 444422, 515555,
        551555, 555155, 555515, 555551, 666666, 1224444
    };

    int nextBeautifulNumber(int n) {
        return *upper_bound(balance.begin(), balance.end(), n);
    }
};

同时也可以使用枚举法，因为我们知道这个题目中可能出现的最大平衡树就是12444，所以只需将n+1到122444所有的数进行枚举，取第一个大于n的平衡树即可，这里需要单独写一个判断平衡树的函数；

class Solution {
public:

    bool isBalance(int num){
        vector<int> count(10);
        int n = 0;
        while(num > 0){
            count[num % 10] ++;
            num /= 10;
        }
        for(int i = 0; i < 10; i ++){
            if(count[i] > 0 && count[i] != i){
                return false;
            }
        }
        return true;
    }
    int nextBeautifulNumber(int n) {
        int i;
        for(i = n + 1; i <= 1224444; ++ i){
            if(isBalance(i)){
                break;
            }
        }
        return i;
    }
};