一、題目
LeetCode-66. 加一
鏈接:https://leetcode-cn.com/problems/plus-one/
難度:簡單
給定一個由整數組成的非空數組所表示的非負整數,在該數的基礎上加一。
最高位數字存放在數組的首位, 數組中每個元素只存儲單個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入:digits = [1,2,3]
輸出:[1,2,4]
解釋:輸入數組表示數字 123。
示例 2:
輸入:digits = [4,3,2,1]
輸出:[4,3,2,2]
解釋:輸入數組表示數字 4321。
示例 3:
輸入:digits = [0]
輸出:[1]
提示:
1 <= digits.length <= 100
0 <= digits[i] <= 9
二、解題思路
根據題意加一,因此所有可能的情況就只有兩種:
- 除 99 之外的數字加一;
- 數字 99。
加一得十進一位個位數為 00 加法運算如果不出現進位,運算就結束了且進位只會是一。
所以只需要判斷有沒有進位并模擬出它的進位方式,如十位數加 11 個位數置為 00,如此循環直到判斷沒有再進位就退出循環返回結果。
同樣的,當出現 9999、999999 之類的數字時,循環到最后也需要進位,出現這種情況時需要手動將它進一位。
三、實現過程
c++
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int len = digits.size();
for(int i = len - 1; i >= 0; i--) {
digits[i]++;
digits[i] %= 10;
if(digits[i]!=0)
return digits;
}
vector<int> tmp(len + 1,0);
tmp[0] = 1;
return tmp;
}
};
PHP
class Solution {
/**
* @param Integer[] $digits
* @return Integer[]
*/
function plusOne($digits) {
$len = count($digits);
for($i = $len - 1; $i >= 0; $i--) {
$digits[$i]++;
$digits[$i] %= 10;
if($digits[$i]!=0){
return $digits;
}
}
$digits[$len + 1] = 0;
$digits[0] = 1;
return $digits;
}
}
JavaScript
/**
* @param {number[]} digits
* @return {number[]}
*/
var plusOne = function(digits) {
const len = digits.length;
for(let i = len - 1; i >= 0; i--) {
digits[i]++;
digits[i] %= 10;
if(digits[i]!=0)
return digits;
}
digits = [...Array(len + 1)].map(_=>0);;
digits[0] = 1;
return digits;
};
四、小結
- 時間復雜度:O(N),其中N是數組nums的長度
- 空間復雜度:O(1)