LeetCode 6303. 分割数组中数字的数位

给你一个正整数数组 nums ，请你返回一个数组 answer ，你需要将 nums 中每个整数进行数位分割后，按照 nums 中出现的 相同顺序 放入答案数组中。

对一个整数进行数位分割，指的是将整数各个数位按原本出现的顺序排列成数组。

比方说，整数 10921 ，分割它的各个数位得到 [1,0,9,2,1] 。

示例 1：

输入：nums = [13,25,83,77]
输出：[1,3,2,5,8,3,7,7]
解释：

• 分割 13 得到 [1,3] 。
• 分割 25 得到 [2,5] 。
• 分割 83 得到 [8,3] 。
• 分割 77 得到 [7,7] 。
  answer = [1,3,2,5,8,3,7,7] 。answer 中的数字分割结果按照原数字在数组中的相同顺序排列。

1 <= nums.length <= 1000
1 <= nums[i] <= 105

解法一：用栈存放每个数字的数位，然后将其加入结果数组：

class Solution {
public:
    vector<int> separateDigits(vector<int>& nums) {
        stack<int> is;
        vector<int> ans;
        for (int i : nums) {
            while (i) {
                is.push(i % 10);
                i /= 10;
            }

            while (!is.empty()) {
                ans.push_back(is.top());
                is.pop();
            }
        }
        
        return ans;
    }
};

如果输入数组nums的长度为n，nums数组中最大数字为m，此算法时间复杂度为O（nlgm），空间复杂度为O（lgm）。

解法二：在解法一的基础上，如果我们手动倒置每个数字在结果数组中的位数，则只需常数空间了：

class Solution {
public:
    vector<int> separateDigits(vector<int>& nums) {
        vector<int> ans;
        for (int i : nums) {
            int numDigitNum = 0;
            while (i) {
                ans.push_back(i % 10);
                i /= 10;
                ++numDigitNum;
            }

            int loopNum = numDigitNum / 2;
            int sz = ans.size();
            while (loopNum) {
                swap(ans[sz - loopNum], ans[sz - numDigitNum + loopNum - 1]);
                --loopNum;
            }
        }
        
        return ans;
    }
};

如果输入数组nums的长度为n，nums数组中最大数字为m，此算法时间复杂度为O（nlgm），空间复杂度为O（1）。