LeetCode刷题（12）～加一

题目描述

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

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

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

示例 1:

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

示例 2:

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

解答

提交代码

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
       int len=digits.size();
       for(int i=len-1;i>=0;--i)
       {
           ++digits[i];
           digits[i]%=10;
           if(digits[i]!=0) return digits;
       }
       digits.resize(len+1,0);
       digits[0]=1;
       return digits;
    }
};

运行结果

c++测试代码

#include<iostream>
#include<vector>
using namespace std;
 vector<int> plusOne(vector<int>& digits) {
       int len=digits.size();
       for(int i=len-1;i>=0;--i)
       {
           ++digits[i];
           digits[i]%=10;
           if(digits[i]!=0) return digits;
       }
       digits.resize(len+1,0);
       digits[0]=1;
       return digits;
    }
int main()
{
    vector<int> a;
    a.push_back(1);
    a.push_back(2);
    a.push_back(3);
    vector<int> b=plusOne(a);
    for(int i=0;i<b.size();++i)
    cout<<b[i];
    cout<<endl;
    
    return 0;
}

思路
        ​​​传送门