力扣66. 加一

模拟

• 思路：
  • 关注数组末尾出现 9 的个数；
  • 找到不为 9 的下标，将对应的值 + 1，然后将其后的数值全部置 0 ；
  • 如果全部为 9，则数组扩展 1 位，将最高位（第一个元素）置 1 ，其余全部为 0；
• 示例：
  • [1, 2, 8] 没有 9 ，直接将元素（倒序）加 1 即可，[1, 2, 9]
  • [1, 4, 8, 9, 9]，将倒序第一个非 9 的数值加 1， [1, 4, 9, 9, 9]，然后将其后元素置零 [1, 4 , 9, 0, 0]
  • [9, 9, 9, 9] 全部为 9，数组扩展一位，第一个元素为 1，其余为 0，[1, 0, 0, 0, 0]

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int n = digits.size();
        for (int i = n - 1; i >= 0; --i) {
            if (digits[i] != 9) {
                ++digits[i];
                for (int j = i +1; j < n; ++j) {
                    digits[j] = 0;
                }

                return digits;
            }
        }

        std::vector<int> ans(n + 1);
        ans[0] = 1;

        return ans;
    }
};

---