LeetCode 989. 数组形式的整数加法

LeetCode 989. 数组形式的整数加法

描述

整数的 数组形式 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

题解

class Solution {
    public List<Integer> addToArrayForm(int[] num, int k) {
        List<Integer> res = new ArrayList<>();
        // 进位
        int carry = 0;
        // 数组形式整数的最后一位索引
        int index = num.length - 1;
        // 相加
        while (index >= 0 || k > 0) {
            // 与上一步的进位相加
            int sum = carry;
            // 与数组当前位相加
            if (index >= 0) {
                sum += num[index];
                index--;
            }
            // 与当前位整数相加
            if (k > 0) {
                sum += k % 10;
                k /= 10;
            }
            // 对10取余
            res.add(0, sum % 10);
            // 更新进位
            carry = sum / 10;
        }
        // 进位
        if (carry > 0) {
            res.add(0, carry);
        }
        // 返回结果
        return res;
    }
}