题目
题目来源
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
思路
首先判断,如果是0,返回1
若不是0,找到最后一位并对其自增1,(该过程使用count进行位数的移动),如果自增大于等于10,则产生进位并使用while语句进行判断,直至count位小于等于9,若count已经到了0的位置还需要进位,则在vector的首位增加一位"1"。
相关思考
一开始没看懂题目,以为是直接给最后一个元素加1(当时非常疑惑为什么会有这种题),后来才发现类似将数组看成一个数字,然后如果产生进位的话需要将前一位也加一,以此类推。(就像评论说的,如果给一个输入[9]输出[1,0]的例子就容易理解多了)
如果进到首位还需要+1的话,需要使用vector在首个元素之前的插入语句 digits.insert(digits.begin(),1)
(由此可见 vector的int数组真的比普通的int数组方便!)
代码(C++/原创)
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
if(digits[0]==0)
{
digits[0]=1;
return digits;
}
else
{
int count=digits.size()-1;
digits[count]+=1;
while(digits[count]>=10)
{
if(count-1>=0)
{
digits[count-1]+=1;
digits[count]-=10;
count--;
}else
{
digits[count]-=10;
digits.insert(digits.begin(), 1);
}
}
return digits;
}
}
};