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();}}}};