leetcode双周赛121最后一题,错了好多次,有点扣脚,记个自己的模板。
题目描述
从start至finish,每个数位都不能超过limit,且数的后缀为s(保证s的每个数位都不超过limit),求一共有多少个这样的数。
大致思路:
只要求出0~finish 有符合要求的数以及0~start-1有多少这样的数,然后减一减就行了,具体怎么算,详情看代码。
代码如下:
using ll = long long;
class Solution {
private:
int limit;
long long countIt(long long x, string s)
{
// 将x做拆分
vector<int> arr;
while (x > 0)
{
arr.push_back(x % 10);
x /= 10;
}
// 如果x的长度比s还小,直接返回0
if (arr.size() < s.size()) return 0;
int n = arr.size(), m = s.size();
ll ans = 0;
for (int i = n - 1; i >= 0; --i)
{
if (i + 1 > m)
{
// 位数大于后缀长度时
if (arr[i] == limit)
{
//如果当前位等于limit,那么当前位为0~limit-1时,后置位可以只在limit的限制下取,limit后置位有i - m位,个数即为pow(limit + 1, i - m),再乘以limit;当前位为limit时,继续看下一位。
ans += pow(limit + 1, i - m) * limit;
}
else if (arr[i] < limit)
{
// 如果当前位小于limit,那么当前位为0~arr[i] - 1时可以为所欲为;当前位为arr[i]时,继续看下一位。
ans += pow(limit + 1, i - m) * arr[i];
}
else if (arr[i] > limit)
{
// 如果大于limit,那么当前位为0~limit都可以为所欲为,剩下的不用看了
ans += pow(limit + 1, i + 1 - m);
break;
}
}
else
{
// 位数小于等于后缀长度时
if (arr[i] > s[m - 1 - i] - '0')
{
// 当前位大于后缀的当前位,那么后缀必定可以取到+1,退出循环
ans += 1;
break;
}
else if (arr[i] == s[m - 1 - i] - '0')
{
// 当前位等于后缀的当前位,如果已经是个位时+1,否则继续看下一位
if (i == 0)
{
ans += 1;
}
}
else if (arr[i] < s[m - 1 - i] - '0')
{
// 当前位小于后缀的当前位,那么必取不到后缀了,退出循环
break;
}
}
}
return ans;
}
public:
long long numberOfPowerfulInt(long long start, long long finish, int limit, string s) {
this->limit = limit;
return countIt(finish, s) - countIt(start - 1, s);
}
};