算法练习:数组形式的整数加法

题目:

对于非负整数 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

来源：力扣（LeetCode）

逐位相加法
就是把K的每一位分别和数组A对应的元素相加，如果相加结果sum小于10，则直接存入集合，若大于10，则先将K/10后加1（相当于进位），然后把sum-10后存入集合，每一位都存入后对集合进行逆序可得最终结果。

按照这个思路写如下代码：

public List<Integer> addToArrayForm(int[] A, int K) {
       int a,sum;
		List<Integer> list=new ArrayList<Integer>();
        for (int i =  A.length-1; i>=0; i--) {
			a=K%10;
			sum=A[i]+a;
			K/=10;
			if (sum<10) {
				list.add(sum);
			}else {
				K++;
				sum-=10;
				list.add(sum);
			}
		}
        //逆序
        Collections.reverse(list);
        return list;
    }

但是运行示例3后结果为[0,2,1]，检查后发现是因为for循环只有A的长度个，而A组成的数和K相加后得到的数的位数大于A的长度个，因此最后一位没有被加进集合。

可以在最后增加一个for循环：

for (;K>0; K/=10) {
	sum=K%10;
	list.add(sum);
}

如果K>0，则意味着还有元素没有被添加进集合，直接把K的每一位数按顺序添加进去。