989. 数组形式的整数加法

package com.heu.wsq.leetcode.arr;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * 989. 数组形式的整数加法
 * @author wsq
 * @date 2021/1/22
 * 对于非负整数 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
 *
 * 链接：https://leetcode-cn.com/problems/add-to-array-form-of-integer
 */
public class AddToArrayForm {
    public List<Integer> addToArrayForm(int[] A, int K) {
        int n = A.length;
        List<Integer> res = new ArrayList<>();
        // 求k的大小
        int add = 0;
        int i = n - 1;
        while(i >= 0 || K != 0){
            int tmp = K % 10;
            int num = i >= 0 ? A[i] : 0;
            tmp = num + tmp + add;
            add = 0;
            if(tmp >= 10){
                add = tmp / 10;
                tmp = tmp % 10;
            }
            res.add(tmp);
            K /= 10;
            i--;
        }
        if(add != 0){
            res.add(add);
        }
        Collections.reverse(res);
        return res;
    }
}