LeetCode: 66. 加一

91 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第9天,点击查看活动详情

66. 加一

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/pl…

给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。

最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外,这个整数不会以零开头。

示例 1:

输入:digits = [1,2,3] 输出:[1,2,4] 解释:输入数组表示数字 123。

示例 2:

输入:digits = [4,3,2,1] 输出:[4,3,2,2] 解释:输入数组表示数字 4321。

示例 3:

输入:digits = [0] 输出:[1]

提示:

1 <= digits.length <= 100 0 <= digits[i] <= 9

解法

  • 方法1-调包:python int 将str转换成整数, list将str转换成列表

  • python

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        data = str(int(''.join([str(i) for i in digits])) + 1)
        data_s = list(data)
        return [int(i) for i in data_s]
  • 方法2-模拟加法,从尾部往前加1,分析末尾9的情况

    • 如果 digits 的末尾没有 9,例如 [1,2,3],那么我们直接将末尾的数加一,得到[1,2,4] 并返回;
    • 如果 digits 的末尾有若干个 9,例如 [1,2,3,9,9],那么我们只需要找出从末尾开始的第一个不为 9 的元素,即 3,将该元素加一,得到 [1, 2, 4, 9, 9]。随后将末尾的 9 全部置零,得到 [1, 2, 4, 0, 0]并返回
    • 如果digits 的所有元素都是 9,例如 [9, 9, 9, 9, 9],那么答案为 [1, 0, 0, 0, 0, 0]。只需要构造一个长度比digits 多 1 的新数组,将首元素置为 1,其余元素置为 0 即可
  • python

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        n = len(digits)
        for i in range(n-1, -1, -1):
            if digits[i] != 9:
                digits[i] += 1
                for j in range(i+1, n):
                    digits[j] = 0
                return digits
        return [1] + [0] * n
  • c++
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] += 1;
                for(int j=i+1; j<n; j++)
                {
                    digits[j] = 0;
                }
            return digits;
            }
        }
        vector<int> ans(n + 1);
        ans[0] = 1;
        return ans;
    }
};

复杂度分析

  • 时间复杂度:O(n)
  • 空间复杂度:O(1)