【leetcode之旅】陣列- 66. 加一問題
阿新 • • 發佈:2019-02-17
給定一個非負整數組成的非空陣列,在該數的基礎上加一,返回一個新的陣列。
最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入陣列表示數字 123。
示例 2:
輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。
public class PlusOne { public static void main(String[] args) { int[] nums = {9,8,9,9}; int[] results = plusOne1(nums); for (int i = 0; i < results.length; i++) { System.out.print(results[i]); } } /** * 方法一: * @param digits * @return */ public static int[] plusOne1(int[] digits){ int len = digits.length; for (int i = len - 1; i >= 0; i--) { if (digits[i] == 9){ digits[i] = 0; }else{ digits[i]++; return digits; } } int[] results = new int[len + 1]; results[0] = 1; return results; } /** * 方法二: * @param digits * @return */ public static int[] plusOne2(int[] digits){ int plus = 1; for (int i = digits.length - 1; i >= 0; i--) { if (plus == 0) { return digits; } int tmp = digits[i] + plus; //若進位則為1,,否則為0 plus = tmp / 10; //若進位則為0,,否則加一 digits[i] = tmp % 10; } //如果每位數都為9,則需要擴容 if (plus != 0) { int[] result = new int[digits.length + 1]; result[0] = 1; return result; } return digits; } }