给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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
思路:
当我们对数组 digits 加一时,我们只需要关注 digits的末尾出现了多少个 999 即可。我们可以考虑如下的三种情况:
1、如果的末尾没有 999,例如 [1,2,3][1, 2, 3][1,2,3],那么我们直接将末尾的数加一,得到 [1,2,4][1, 2, 4][1,2,4] 并返回;
2、如果 digits 的末尾有若干个 999,例如 [1,2,3,9,9][1, 2, 3, 9, 9][1,2,3,9,9],那么我们只需要找出从末尾开始的第一个不为 999 的元素,即 333,将该元素加一,得到 [1,2,4,9,9][1, 2, 4, 9, 9][1,2,4,9,9]。随后将末尾的 999 全部置零,得到 [1,2,4,0,0][1, 2, 4, 0, 0][1,2,4,0,0] 并返回。
3、如果 digits的所有元素都是 999,例如 [9,9,9,9,9][9, 9, 9, 9, 9][9,9,9,9,9],那么答案为 [1,0,0,0,0,0][1, 0, 0, 0, 0, 0][1,0,0,0,0,0]。我们只需要构造一个长度比 digits多 111 的新数组,将首元素置为 111,其余元素置为 000 即可。
我的思路与官方思路一模一样,只是没有官方的简洁但核心代码几乎一致,这里还是用官方代码来呈现:
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]++;
for (int j = i + 1; j < n; j++)
{
digits[j] = 0;
}
return digits;
}
}
// digits 中所有的元素均为 9
vector<int> ans(n + 1);
ans[0] = 1;
return ans;
}
};
这里我也有一个问题,题解中给出的是--i、++digits,为什么比i++和dights++效率更高?