leetcode基础算法 数组篇:数组加一

给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。

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

示例 1:

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

示例 2:

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

思路:就是用数组表示非负整数,将这个非负整数加1,用数组返回。

大致分了三种情况:

1、从个位起,没有9的话,直接在尾数加1.例如1234,123.

2、有9但不全为9,例如129,1239,把9那位变0,下一位+1.

3、所有元素都是9,例如9,999这样的,就新建一个数组,长度为原数组长度+1,第一位置为1就好.例如999->10000.

代码如下:

class Solution {
public:
    vector plusOne(vector& digits) {
        for(int i=digits.size()-1;i>=0;i--)   //从最后一个元素向前遍历
            if(digits[i]==9)              //如果是9就置为0
                digits[i]=0;
             else 
             {
                 digits[i]++;            //不是9就++,数组加一完成,返回更改后的原数组
                 return digits;
             }
         
            vectordigits2(digits.size()+1);     //全为9的情况,就新建一个数组
               digits2[0]=1;
                return digits2;          
    }
};

 

你可能感兴趣的:(leetcode基础算法)