整数的 数组形式 num
是按照从左到右的顺序表示其数字的数组。
- 例如,对于
num = 1321
,数组形式是[1,3,2,1]
。
给定 num
,整数的 数组形式 ,和整数 k
,返回 整数 num + k
的 数组形式 。
示例 1:
输入:num = [1,2,0,0], k = 34 输出:[1,2,3,4] 解释:1200 + 34 = 1234
示例 2:
输入:num = [2,7,4], k = 181 输出:[4,5,5] 解释:274 + 181 = 455
示例 3:
输入:num = [2,1,5], k = 806 输出:[1,0,2,1] 解释:215 + 806 = 1021
提示:
1 <= num.length <= 104
0 <= num[i] <= 9
num
不包含任何前导零,除了零本身1 <= k <= 104
数组从尾到头遍历,分别与 K
中的每一位相加,进位保存在 carry
中,不进位和则添加到结果列表中。最后逆序结果列表即可。
class Solution:
def addToArrayForm(self, A: List[int], K: int) -> List[int]:
n = len(A) - 1
carry, res = 0, []
while n >= 0 or K != 0 or carry != 0:
carry += (0 if n < 0 else A[n]) + (K % 10)
res.append(carry % 10)
K //= 10
carry //= 10
n -= 1
return res[::-1]
class Solution {
public List<Integer> addToArrayForm(int[] A, int K) {
int n = A.length - 1;
List<Integer> res = new ArrayList<>();
int carry = 0;
while (n >= 0 || K != 0 || carry != 0) {
carry += (n < 0 ? 0 : A[n]) + (K % 10);
res.add(carry % 10);
K /= 10;
carry /= 10;
--n;
}
Collections.reverse(res);
return res;
}
}
function addToArrayForm(num: number[], k: number): number[] {
let arr2 = [...String(k)].map(Number);
let ans = [];
let sum = 0;
while (num.length || arr2.length || sum) {
let a = num.pop() || 0,
b = arr2.pop() || 0;
sum += a + b;
ans.unshift(sum % 10);
sum = Math.floor(sum / 10);
}
return ans;
}