leetcode--加一(PlusOne)--java实现

题目：加一

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

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

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

package stone.scu.cd;
//题目：给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

public class OnePlus {
	public static void main(String[] args) {
		int[] digits = {9,9,9};
		int[] res = new OnePlus().plusOne(digits);
		for (int i = 0; i < res.length; i++) {
			System.out.print(res[i] + " ");
		}
	}
	
	public int[] plusOne(int[] digits) {
		int carry = 1;  //进位标志,初始化为1（加1）
		int i = digits.length - 1;
		while (i >= 0 && carry > 0) {  //最高位存放在首位，所以从后往前遍历
			int sum = digits[i] + carry;
			carry = sum / 10;          //保存下一个carry值：sum为10 --> carry=1，表示进位
			digits[i] = sum % 10;      //sum<10 --> 不进位，为加1后的数值
			--i;
		}
		//最高位也需要进位（i减至0时，carry仍为1）
		if (carry > 0) {
			int[] newArray = new int[digits.length + 1];   //新建数组，长度+1
			System.arraycopy(digits, 0, newArray, 1, digits.length);  //将digits数组(从0开始)拷贝digits.length个元素到newArray数组(从1开始)
			newArray[0] = carry;  //首位为1
			return newArray;
		}
		return digits;
	}
}