题目描述
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例:
输入: 输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
思路
1.我们可以从数组的最后一位开始加一,加一后将值与10进行mod运算,若取模后等于0,说明要进行进位操作,继续进行循环即可;若不等于0,说明不需要进位,直接返回即可。
2.若我们从循环出来后还未返回结果,说明还需要进位操作,我们初始化一个长度比之前数组大一的新数组,然后将数组的第一个位置赋值为1即可,因为数组的默认值是0,是与原数组相同的。
Java代码实现
public int[] plusOne(int[] digits) { int len = digits.length; for(int i=len-1;i>=0;i--){ digits[i] = digits[i] + 1; digits[i] = digits[i]%10; if(digits[i] != 0) return digits; } digits = new int[len + 1]; digits[0] = 1; return digits; }
Golang代码实现
func plusOne(digits []int) []int { if len(digits) == 0{ return digits } length := len(digits) for i:= length-1;i>=0;i--{ cur := digits[i]+1 digits[i] = cur%10 if digits[i] != 0{ return digits } } digits = make([]int,length+1) digits[0] = 1 return digits; }