【LeetCode刷题记录】简单篇-66-加一

【题目描述】

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

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

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

---

【测试用例】

示例1：

        输入：digits = [1,2,3]

        输出：[1,2,4]

示例2：

        输入：digits = [4,3,2,1]

        输出：[4,3,2,2]

示例3：

        输入：digits = [0]

        输出：[1]

---

【思路分析】

先说一下我最开始的做法吧。依旧是大部分人第一次见到题会想到的办法。直接把数组里面的数拿出来组成非负整数，然后加一，再按位依次放入新数组中，最后返回该数组。

提交后报错溢出了

它的测试用例9876543210，在代码中会溢出，所以我把int改成了long long，这个用例通过了，但是又出现了新的溢出用例

好家伙，我算是看明白了，这道题就是不准你用这种笨方法来做。

那就换一个方法。我们知道，一个非负整数加1，分进位和不进位两种情况。不进位的情况很简单，直接末位加1就行；进位的情况即末位为9，加1后本位变0再向前进位，这是唯一的情况，也就是说所有的进位都按这样的方法计算，那么在代码中就可以用一个循环体解决。

具体来说，用一个count变量存储数组中逆序连续的9，因为9会进位，不是9就一定不会发生进位，所以只要逆序找到第一个不为9的数，将其加1，这个数之前的所有数不变，之后的数全部置0即可。但是还有一种特殊情况：数组中的数全为9，比如[9,9,9]，此时加1后数组会增加一位，该位为1，其余位全为0，即[1,0,0,0]。这在代码中很好实现。

---

【参考代码】

C实现

#include <stdio.h>
#include <stdlib.h>

//easy-66-加一
int* plusOne(int* digits, int digitsSize, int* returnSize);

int main(){
	int digits[] = {1,2,3};
	int digitsSize = sizeof(digits)/sizeof(digits[0]);
	int* returnSize;
	int* res = plusOne(digits, digitsSize, returnSize);
	int i;
	for(i=0;i<*returnSize;++i){
		printf("%d ", res[i]);
	}
	return 0;
}

int* plusOne(int* digits, int digitsSize, int* returnSize) {
    int i, count=0;
    *returnSize = digitsSize;
	for(i=digitsSize-1;i>=0;--i){
		if(digits[i] == 9){
			count++;
		}else{
			digits[i] += 1;
			break;
		}
	} 
	if(count == digitsSize){
		int* res = (int*)malloc(sizeof(int)*(digitsSize+1));
		*returnSize = digitsSize + 1;
		res[0] = 1;
		for(i=1;i<digitsSize+1;i++){
			res[i] = 0;
		}
		return res;
	}
	i = digitsSize-1;
	while(count>0){
		digits[digitsSize-count] = 0;
		count--;
	}
	return digits;
}

C++实现

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

//easy-66-加一
class Solution {
public:
    vector<int> plusOne(vector<int>& digits);
};

vector<int> Solution::plusOne(vector<int>& digits){
	int i, count=0;
	for(i=digits.size()-1;i>=0;--i){
		if(digits[i] == 9){
			count++;
		}else{
			digits[i] += 1;
			break;
		}
	}
	if(count == digits.size()){
		vector<int> res(digits.size()+1, 0);
		res[0] = 1;
		return res;
	}
	while(count > 0){
		digits[digits.size()-count] = 0;
		count--;
	}
	return digits;
}

int main(){
	vector<int> digits = {1,2,3};
	Solution sol;
	vector<int> res = sol.plusOne(digits);
	int i;
	for(i=0;i<res.size();++i){
		cout<<res[i]<<" ";
	}
	return 0;
}