[Leetcode] 0066. 加一

66. 加一

题目描述

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

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

你可以假设除了整数 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

解法

方法一:找出最长的后缀 9

思路

当我们对数组 \(\textit{digits}\) 加一时,我们只需要关注 \(\textit{digits}\) 的末尾出现了多少个 \(9\) 即可。我们可以考虑如下的三种情况:

如果 \(\textit{digits}\) 的末尾没有 \(9\),例如 \([1, 2, 3]\),那么我们直接将末尾的数加一,得到 \([1, 2, 4]\) 并返回;

如果 \(\textit{digits}\) 的末尾有若干个 \(9\),例如 \([1, 2, 3, 9, 9]\),那么我们只需要找出从末尾开始的第一个不为 \(9\) 的元素,即 \(3\),将该元素加一,得到 \([1, 2, 4, 9, 9]\)。随后将末尾的 \(9\) 全部置零,得到 \([1, 2, 4, 0, 0]\) 并返回。

如果 \(\textit{digits}\) 的所有元素都是 \(9\),例如 \([9, 9, 9, 9, 9]\),那么答案为 \([1, 0, 0, 0, 0, 0]\)。我们只需要构造一个长度比 \(\textit{digits}\)\(1\) 的新数组,将首元素置为 \(1\),其余元素置为 \(0\) 即可。

算法

们只需要对数组 \(\textit{digits}\) 进行一次逆序遍历,找出第一个不为 \(9\) 的元素,将其加一并将后续所有元素置零即可。如果 \(\textit{digits}\) 中所有的元素均为 \(9\),那么对应着「思路」部分的第三种情况,我们需要返回一个新的数组。

复杂度分析

时间复杂度:\(O(n)\),其中 \(n\) 是数组 \(\textit{digits}\) 的长度。

空间复杂度:\(O(1)\)。返回值不计入空间复杂度。

Python3

# class Solution:
#     def plusOne(self, digits: List[int]) -> List[int]:
#         digits = [str(i) for i in digits]
#         num = int("".join(digits)) + 1
#         a = str(num)
#         list1=[]
#         for i in a:
#             list1.append(int(i))
#         return list1

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];
                for(int j=i+1;j<n;++j)
                    digits[j] =0;
                return digits;
            }
        }
        vector<int> ans(n+1);
        ans[0] = 1;
        return ans;
    }
};

作者:野哥李
微信公众号:AI算法学习社
欢迎任何形式的转载,但请务必注明出处。
限于本人水平,如果文章和代码有表述不当之处,还请不吝赐教。
本文章不做任何商业用途,仅作为自学所用,文章后面会有参考链接,我可能会复制原作者的话,如果介意,我会修改或者删除。

posted @   野哥李  阅读(29)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 如何使用 Uni-app 实现视频聊天(源码,支持安卓、iOS)
· C# 集成 DeepSeek 模型实现 AI 私有化(本地部署与 API 调用教程)
点击右上角即可分享
微信分享提示