LeetCode-66. Plus One

最新推荐文章于 2022-04-21 15:55:14 发布

原创最新推荐文章于 2022-04-21 15:55:14 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode题解专栏收录该内容

85 篇文章

订阅专栏

本文介绍了一种针对整数加一操作的算法优化方法。该方法通过遍历数组从末尾开始检查并修改每一位数字，实现整数加一的功能。优化后的算法将最坏情况下的时间复杂度从O(2N)降低到O(N)，显著提高了执行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a non-empty array of digits representing a non-negative integer, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

Example 1:

Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.

Example 2:

Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.

参考代码：

1.最坏情况下，时间复杂度为O（2N）。digits的insert函数要执行一遍数组。

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int carry=1,flag=1;
        for(int i=digits.size()-1;i>=0&&flag;--i)
        {
            digits[i]+=carry;
            carry=0;
            flag=0;
            if(digits[i]>=10) {
                digits[i]=0;
                carry=1;
                flag=1;
            }
        }
        if(carry==1) digits.insert(digits.begin(),1);
    return digits;
    }
};

优化：

1.最坏情况下时间复杂度O（N）。

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