第 264 场周赛Cider T2.2048.下一个更大的数值平衡数

解题思路

从n+1开始遍历,进行判断当前数字是否符合,这个过程需要剪枝

代码

class Solution {
    public int nextBeautifulNumber(int n) {
        for (int i=n+1;i<=Integer.MAX_VALUE;i++){
            if (check(i)){
                return i;
            }
        }
        return n;
    }
    public boolean check(int n){
        char[] chars = Integer.toString(n).toCharArray();
        Set<Integer> set = new HashSet<>();
        int[] mem = new int[10];
        for (int i=0;i<chars.length;i++){
            //不能出现0
            if (mem[chars[i]-48]+1 > (chars[i]-48)){
                return false;
            }else {
                set.add(chars[i]-48);
                mem[chars[i]-48]++;
            }
        }
        for (int key : set){
            if (key != mem[key]){
                return false;
            }
        }
        return true;
    }
}