加一
概述:给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。你可以假设除了整数 0 之外,这个整数不会以零开头。
输入:digits = [1,2,3]
输出:[1,2,4]
输入:digits = [4,3,2,1]
输出:[4,3,2,2]
输入:digits = [0]
输出:[1]
方法一:数组逆循环
思路:我们只需要对数组 digits 进行一次逆序遍历,判断最后一位是否为9,有则进入下一次循环,直到找到不为9加上1返回数组,若索引为0的那一位是9,则需要进行数组扩容,我们可以利用System中提供的arraycopy()方法,不需要for循环的方式将现有数组复制到扩容后的数组中。
话不多说,直接上代码!
public static int[] plusOne(int[] digits) {
for (int i = digits.length-1; i >= 0; i--) {
if (digits[i] == 9){
digits[i] = 0;
//增加数组长度
if (i==0){
int [] arr = new int[digits.length+1];
System.arraycopy(digits, 0, arr, 1, digits.length);
arr[0] = 1;
return arr;
}
}else {
digits[i] += 1;
return digits;
}
}
return digits;
}
测试:
public static void main(String[] args) {
int[] digits = {9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9};
int[] ints = plusOne(digits);
System.out.println(Arrays.toString(ints));
}
输出:
arraycopy()方法参数介绍:
public static native void arraycopy(Object src, int srcPos, Object dest, int destPos, int length)
大白话总结:
从源数组src取元素,范围为下标srcPos到srcPos+length-1,取出共length个元素,存放到目标数组中,存放位置为下标destPos到destPos+length-1。简单说,就是数组间的复制。
src 源数组;
srcPos 源数组中的起始位置;
dest 目标数组;
destPos 目标数组中的起始位置;
length 要复制的数组元素的数量
方法二:转成BigDecimal进行计算
注意:这里因为数组的长度是没有限制的,所以基本数据类型及其包装类的范围都是不够的。
class Solution {
public int[] plusOne(int[] digits) {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < digits.length; i++) {
sb.append(digits[i]);
}
BigDecimal b1 = new BigDecimal(sb.toString());
BigDecimal b2 = new BigDecimal("1");
BigDecimal add = b1.add(b2);
String s = add.toString();
digits = new int[s.length()];
for (int i = 0; i < digits.length; i++) {
digits[i] = Character.getNumericValue(s.charAt(i));
}
return digits;
}
}
测试:
public static void main(String[] args) {
int[] digits = {1,2,3};
int[] ints = plusOne2(digits);
System.out.println(Arrays.toString(ints));
}
输出: