力扣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 plusOne(vector& 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 ans(n + 1);
        ans[0] = 1;

        return ans;
    }
};

————————————————————————————

力扣66. 加一_第1张图片

你可能感兴趣的:(力扣实践,leetcode,算法,职场和发展)