start至finish，每个数位不超过limit，数的后缀为s

原创已于 2024-01-07 00:15:43 修改 · 429 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

于 2024-01-07 00:00:06 首次发布

leetcode周赛同时被 2 个专栏收录

8 篇文章

订阅专栏

c++

2 篇文章

订阅专栏

文章介绍了如何解决LeetCode中一道关于计数特定范围内数字的问题，涉及到数位限制和后缀条件。作者提供了一个Solution类的代码片段，通过拆分数字和比较数位来计算符合条件的数的数量。

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