66. 加一
思路:模拟
对数组digits进行一次逆序遍历,找到第一个不为0的元素,加一并将后续所有元素置0即可,如果digits种所有元素均为0,那么需要返回一个新数组。
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
for(int i=digits.size()-1;i>=0;--i){
if(digits[i]!=9){
digits[i]++;
for(int j=i+1;j<digits.size();++j) digits[j]=0;
return digits;
}
}
vector<int> ans(digits.size()+1);
ans[0]=1;
return ans;
}
};
时间复杂度 O(n)
空间复杂度 O(1)