91算法第四期刷题笔记
day01-989. 数组形式的整数加法
题目
对于非负整数 X 而言,X 的数组形式是每位数字按从左到右的顺序形成的数组。例如,如果 X = 1231,那么其数组形式为 [1,2,3,1]。
给定非负整数 X 的数组形式 A,返回整数 X+K 的数组形式。
示例 1:
输入:A = [1,2,0,0], K = 34
输出:[1,2,3,4]
解释:1200 + 34 = 1234
示例 2:
输入:A = [2,7,4], K = 181
输出:[4,5,5]
解释:274 + 181 = 455
示例 3:
输入:A = [2,1,5], K = 806
输出:[1,0,2,1]
解释:215 + 806 = 1021
示例 4:
输入:A = [9,9,9,9,9,9,9,9,9,9], K = 1
输出:[1,0,0,0,0,0,0,0,0,0,0]
解释:9999999999 + 1 = 10000000000
提示:
1 <= A.length <= 10000
0 <= A[i] <= 9
0 <= K <= 10000
如果 A.length > 1,那么 A[0] != 0
**前置知识:**数组的遍历
**标签:**数组
思路
逐位相加
首先新建一个链表,从数组num尾部和K的低位开始反向遍历,依次求和得到对应的sum,根据求和结果判断是否进位(carry=1表示进位,carry=0表示无进位),将sum对10取模后加入到链表头部。
代码
class Solution {
public static List<Integer> addToArrayForm(int[] num, int k) {
LinkedList<Integer> res = new LinkedList<>();
int carry = 0; //进位数,1表示进位,0表示无进位
int sum; //每次求和的值
int n = num.length - 1; //遍历数组时的索引
while (k != 0 || n >= 0){
sum = carry + (k % 10);
if (n >= 0){
sum += num[n];
}
carry = (sum > 9) ? 1:0;
res.addFirst(sum % 10);
n--;
k = k / 10;
}
if (carry == 1){res.addFirst(1);}
return res;
}
}
复杂度分析
- 时间复杂度:O(max(n, k)),其中 n 为数组长度, k为数值K的长度。
- 空间复杂度:O(1)