加一
- 给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入:digits = [1,2,3]
输出:[1,2,4]
解释:输入数组表示数字 123。
解题思路
-
对数组进行处理,将表示的整数加一。由于数组中每个元素都只存储一个数字,需要从数组的末尾开始逐位处理进位问题。
-
如果所有位都处理完之后仍有进位(例如 999 + 1 变成 1000),需要在数组最前面插入一个 1。
Java实现
public class PlusOne {public int[] plusOne(int[] digits) {int n = digits.length;// 从最后一位开始加一for (int i = n - 1; i >= 0; i--) {if (digits[i] < 9) {digits[i]++;return digits; // 直接返回数组}digits[i] = 0; // 处理进位}// 如果所有位都进位了,需要在最前面插入一个1int[] newDigits = new int[n + 1];newDigits[0] = 1;return newDigits;}// 测试用例public static void main(String[] args) {PlusOne solution = new PlusOne();int[] digits1 = {1, 2, 3};int[] digits2 = {4, 3, 2, 1};int[] digits3 = {9, 9, 9};printArray(solution.plusOne(digits1)); // 期望输出: [1, 2, 4]printArray(solution.plusOne(digits2)); // 期望输出: [4, 3, 2, 2]printArray(solution.plusOne(digits3)); // 期望输出: [1, 0, 0, 0]}private static void printArray(int[] array) {for (int num : array) {System.out.print(num + " ");}System.out.println();}
}
时间空间复杂度
- 时间复杂度:O(n),其中 n 是数组的长度。
- 空间复杂度:O(1),除了结果数组外,使用了常数级别的额外空间。结果数组可能需要额外的空间,如果数组所有位都进位了,需要一个新的数组。