LeetCode--数组形式的简单加法

~澄~

于 2021-01-22 10:14:38 发布

阅读量145

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/luoyeliufeng/article/details/112978300

版权

Java 数组整数相加进位算法实现

关键词由CSDN通过智能技术生成

LeetCode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

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

思路

让数组A从低位向高位和K的相应位置相加，用一个flag判断是否进位

Java实现：

import java.util.*;

public class Solution {
    public List<Integer> addToArrayForm(int[] A, int K) {
        List res = new ArrayList();
        int flag = 0;
        for(int i=A.length-1;i>=0;i--)
        {
            int theSum = A[i]+K%10+flag;
            if(theSum>=10)
                flag = 1;
            else
                flag = 0;
            res.add(theSum%10);
            K= K/10;
            if(i==0 && flag==1 && K==0)
                res.add(flag);
        }
        while(K!=0)
        {
            int theSum = K%10+flag;
            if(theSum>=10)
                flag = 1;
            else
                flag = 0;
            res.add(theSum%10);
            K = K/10;
            if(K==0 && flag==1)
                res.add(flag);
        }
        Collections.reverse(res);
        return res;
    }
    public static void main(String[] arg)
    {
        int[] A = {6};
        int K = 809;
        Solution solution = new Solution();
        List res = solution.addToArrayForm(A, K);
        System.out.println(res);
    }
}