Leetcode 989. 数组形式的整数加法 989. Add to Array-Form of Integer

最新推荐文章于 2022-09-29 18:44:24 发布

歌古道

最新推荐文章于 2022-09-29 18:44:24 发布

阅读量303

点赞数

分类专栏： LeetCode c++ 文章标签： Leetcode C++ 989. 数组形式的整数加法

本文链接：https://blog.csdn.net/qq_27901091/article/details/88969484

版权

c++ 同时被 2 个专栏收录

42 篇文章 0 订阅

订阅专栏

LeetCode

39 篇文章 0 订阅

订阅专栏

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

有一些特殊case，比如
[0]
23
或者
[9,9]
991
看是否可以测试通过

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        int count=0;
        for(int i=A.size()-1;i>=0;i--){ //数组从后往前遍历，有K则加
            int a=0;
            if(K>0){
                a=K%10;
                K=K/10;
            }
            int temp=A[i]+a+count;   //进位，A[i]，K的这一位相加
            if(temp<10) {
                A[i]=temp;
                count=0;
            }
            else{
                A[i]=temp%10;
                count=1;
            }  
        }
        while(K>0){       //有可能数组比K短，这个时候就要再计算一下
            int a=K%10;
            K=K/10;
            if(count==1) a=a+1;
            if(a<10) {
                A.insert(A.begin(),a);
                count=0;
            }
            else{
                A.insert(A.begin(),a%10);
                count=1;
            }  
            
        }                
        if(count==1&&K==0) A.insert(A.begin(),1);//最后的进位还有的话，数组最前面插入1
        return A;
    }
};