题目(困难)
一个 k 镜像数字 指的是一个在十进制和 k 进制下从前往后读和从后往前读都一样的 没有前导 0 的 正 整数。
比方说,9 是一个 2 镜像数字。9 在十进制下为 9 ,二进制下为 1001 ,两者从前往后读和从后往前读都一样。
相反地,4 不是一个 2 镜像数字。4 在二进制下为 100 ,从前往后和从后往前读不相同。
给你进制 k 和一个数字 n ,请你返回 k 镜像数字中 最小 的 n 个数 之和 。
示例 1:
输入:k = 2, n = 5
输出:25
解释:
最小的 5 个 2 镜像数字和它们的二进制表示如下:
十进制 二进制
1 1
3 11
5 101
7 111
9 1001
它们的和为 1 + 3 + 5 + 7 + 9 = 25 。
示例 2:
输入:k = 3, n = 7
输出:499
解释:
7 个最小的 3 镜像数字和它们的三进制表示如下:
十进制 三进制
1 1
2 2
4 11
8 22
121 11111
151 12121
212 21212
它们的和为 1 + 2 + 4 + 8 + 121 + 151 + 212 = 499 。
示例 3:
输入:k = 7, n = 17
输出:20379000
解释:17 个最小的 7 镜像数字分别为:
1, 2, 3, 4, 5, 6, 8, 121, 171, 242, 292, 16561, 65656, 2137312, 4602064, 6597956, 6958596
提示:
2 <= k <= 9
1 <= n <= 30
解题思路
参考了大佬找下一个十进制回文数的思路
ispal函数是判断一个字符串是否回文
toK是吧十进制的num转换成k进制的数,并以字符串形式保存,没有转置这一步是因为如果是回文转置的也是回文,只是判断的话不需要reverse;
next是找当前十进制下的下一个回文数,每次找从中间往两边开花,例如10101的下一个就是10201,中间需要考虑进位问题,如果全是9,那他下一位就是num+2,例如99->101,99999->100001
代码
class Solution {
public:
bool ispal(string s) {
int n = s.size();
for(int i = 0; i < n / 2; ++i) {
if(s[i] != s[n-1-i]) return false;
}
return true;
}
string toK(long long num, int k) {
long long tmp = num;
string ans = "";
while(tmp) {
ans.push_back(tmp % k);
tmp /= k;
}
return ans;
}
long long next(long long num) {
string s = to_string(num);
int n = s.size();
for(int i= n/2; i >= 0; i--){
if(s[i]!='9') { //如果只是中间有9
s[i]++;
if(n-1-i != i){ //偶数长度下保证两边同时增加
s[n-1-i]++;
}
for(int j = i+1; j <= n/2; j++) { //中间的9全进位
s[j] = '0';
s[n-1-j] = '0';
}
return stoll(s);
}
}
//循环完就说明是全9阵容
return num + 2;
}
long long kMirror(int k, int n) {
long long ans = 1;
int cnt = 1;
long long num = 1;
while(cnt < n) {
num = next(num);
if(ispal(toK(num, k))) {
ans += num;
cnt++;
}
}
return ans;
}
};