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

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