给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
思路
标签:数组遍历
这道题需要整理出来有哪几种情况,在进行处理会更舒服
- 末位无进位,则末位加一即可,因为末位无进位,前面也不可能产生进位,比如 45 => 46
- 末位有进位,在中间位置进位停止,则需要找到进位的典型标志,即为当前位 %10后为 0,则前一位加 1,直到不为 0 为止,比如 499 => 500
- 末位有进位,并且一直进位到最前方导致结果多出一位,对于这种情况,需要在第 2 种情况遍历结束的基础上,进行单独处理,比如 999 => 1000
//Java
class Solution { public int[] plusOne(int[] digits) { int len = digits.length; for(int i = len - 1; i >= 0; i--) { digits[i]++; digits[i] %= 10; if(digits[i]!=0) return digits; } digits = new int[len + 1]; digits[0] = 1; return digits; } }
//C语言
int* plusOne(int* digits, int digitsSize, int* returnSize){ for(int i=digitsSize-1;i>=0;--i) { digits[i] = (digits[i]+1)%10; if(digits[i]!=0) break; } *returnSize = digitsSize; if(digits[0]==0) { int *result = (int*)malloc(sizeof(int)*(digitsSize+1)); result[0] = 1; for(int k=1;k<digitsSize+1;++k) result[k] = 0; *returnSize = digitsSize+1; return result; } else return digits; }