LeetCode 902.最大为N的数字组合

模拟手算。

只要是位数小于n的组合一定可以

当位数等于n的时候就要分开开始处理。

从最高位开始确定，对于n的第i位 num[i]存在两种情况。一种为digits中存在等于num[i]的数，第二种为不存在

对两种情况进行分开处理

class Solution {
public:
    int atMostNGivenDigitSet(vector<string>& digits, int n) {
        vector<int> nums;
        int temp = n;
        while(temp)
        {
            nums.push_back(temp % 10);
            temp /= 10;
        }
        
        //位数小于n的情况
        int res = 0;
        temp = 1;
        for(int i = 0; i < nums.size() - 1; i++)
        {
            temp *= digits.size();
            res += temp;
        }

        // 位数等于n
        for (int i = nums.size() - 1; i >= 0; i--)
        {
            // 如果n的这一位小于最小的数，证明不存在这一类型的情况
            if (digits[0][0] - '0' > nums[i])
                return res;
            
            // 从n高位往后进行处理，分为两种情况
            // 1. digits存在等于nums[i]的数，这种情况要再去看下一位，即当nums[i]确定的情况下 nums[i - 1]可能的情况
            // 2. digits不存在等于nums[i]的数，直接返回
            for (int j = digits.size() - 1; j >= 0; j--)
            {
                if (digits[j][0] - '0' < nums[i])
                {
                    temp = 1;
                    for (int k = 0; k < i; k++)
                        temp *= digits.size();
                    res += temp * (j + 1);
                    
                    if (j == digits.size() - 1 || digits[j + 1][0] - '0' > nums[i])
                        return res;
                    break;
                }
            }
        }
        for(int i = 0; i < digits.size(); i++)
        {
            if(digits[i][0] - '0' == nums[0])
                return res + 1;
        }
        return res;
    }
};