难度:Easy
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。 最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。 你可以假设除了整数 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 flag=1;int set=0;int len=digits.size();while(flag){--len;if(digits[len]!=9){digits[len]+=1;flag=0;}else{digits[len]=0;if(len==0){set=1;break;}}}cout<<set<<endl;if(set){len=digits.size();vector<int>tmp(len+1,0);tmp[0]=1;for(int i=1;i<=len;i++)tmp[i]=digits[i-1];return tmp;}return digits;}};