Leetcode-66.加一

题目背景

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

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

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

示例 1：

输入：digits = [1,2,3] 输出：[1,2,4] 解释：输入数组表示数字 123。 示例 2：

输入：digits = [4,3,2,1] 输出：[4,3,2,2] 解释：输入数组表示数字 4321。 示例 3：

输入：digits = [0] 输出：[1]

提示：

1 <= digits.length <= 100 0 <= digits[i] <= 9

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一

#include<iostream>
#include<vector> 
using namespace std;

vector<int> plusOne(vector<int>& digits) {
	int len = digits.size();
	if(digits[len-1]!=9){
		digits[len-1] +=1;
	}else{
		int count=0;
		for(int i=len-1;i>=0;i--){
			if(digits[i]==9){
				digits[i] = 0;
				count++;
			}else{
				break;
			}
		}
		if(count<=len-1)
			digits[len-count-1] +=1;
		else
			digits.insert(digits.begin(),1);
	}
	return digits;
}

int main(){
	vector<int> indig;
	int num;
	while(cin>>num){
		indig.push_back(num);
		if(cin.get()=='\n')
			break;
	}
	
	vector<int> outdig;
	outdig = plusOne(indig);
	for(int i=0;i<outdig.size();i++) 
		cout<<outdig[i]<<" ";
	return 0;
}