开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第9天,点击查看活动详情
66. 加一
来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/pl…
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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
解法
-
方法1-调包:python int 将str转换成整数, list将str转换成列表
-
python
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
data = str(int(''.join([str(i) for i in digits])) + 1)
data_s = list(data)
return [int(i) for i in data_s]
-
方法2-模拟加法,从尾部往前加1,分析末尾9的情况
- 如果 digits 的末尾没有 9,例如 [1,2,3],那么我们直接将末尾的数加一,得到[1,2,4] 并返回;
- 如果 digits 的末尾有若干个 9,例如 [1,2,3,9,9],那么我们只需要找出从末尾开始的第一个不为 9 的元素,即 3,将该元素加一,得到 [1, 2, 4, 9, 9]。随后将末尾的 9 全部置零,得到 [1, 2, 4, 0, 0]并返回
- 如果digits 的所有元素都是 9,例如 [9, 9, 9, 9, 9],那么答案为 [1, 0, 0, 0, 0, 0]。只需要构造一个长度比digits 多 1 的新数组,将首元素置为 1,其余元素置为 0 即可
-
python
class Solution:
def plusOne(self, digits: List[int]) -> List[int]:
n = len(digits)
for i in range(n-1, -1, -1):
if digits[i] != 9:
digits[i] += 1
for j in range(i+1, n):
digits[j] = 0
return digits
return [1] + [0] * n
- c++
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int n = digits.size();
for(int i=n-1; i>=0; i--)
{
if(digits[i] != 9)
{
digits[i] += 1;
for(int j=i+1; j<n; j++)
{
digits[j] = 0;
}
return digits;
}
}
vector<int> ans(n + 1);
ans[0] = 1;
return ans;
}
};
复杂度分析
- 时间复杂度:O(n)
- 空间复杂度:O(1)