LeetCode-989:数组形式的整数加法(图文并茂)

题目描述

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

题解

该题的思想其实很简单，就是整数 k 的个位与数组A的 A[numSize-1]，还有上一个结果的进位进行相加，如果大于10，则产生进位，与下个相加结合，这样一直进行循环，直到数组 A 或 整数K 走完即可。

初始情况

①首先求出整数K的位数Ksize和数组A的个数Asize,然后将两者进行比较，用较大的数作为新数组的长度(这里要开辟一个新的数组)，并且为了防止首位相加产生进位无处存放的情况，应再多开一个空间。

②然后再用两个指针 ai 和 tmp 分别指向数组A的末尾和整数K的个位，设立进位指针next = 0，作为进位的标志，将 ai + tmp + next 的结果放入Arr中。

③循环②的步骤，直到指针 tmp 或 ai 的指针走完，得到Arr数组

④这时候会发现数组Arr结果与原结果是相反的，所以对齐进行逆置即可得最终结果

题解代码

int* addToArrayForm(int* A, int ASize, int K, int* returnSize){
    int num = K;
    //k的大小
    int kSize = 0;
    while(num){
        num /= 10;
        kSize++;
    }
    //求出K和A中的较大值，+1是为了防止进位数组长度不够的情况
    int n = kSize > ASize ? kSize+1 : ASize+1;
    int* Arr = (int* )malloc(sizeof(int)*n);

   //进位指针
    int next = 0;
    //数组指针，从后向前
    int ai = ASize-1;
    //arr数组的指针
    int arri = 0;
    //求出k的个位，并与A中的末尾相加
    while(K > 0 || ai >= 0){
        //取个位
        int tmp = K % 10;
        K = K / 10;
        int Anum = 0;
        if(ai>=0){
            Anum = A[ai--];
        }

        Arr[arri] = Anum + tmp + next;
        
        //如果结果大于10，则进位
        if(Arr[arri] >= 10){
            Arr[arri] -= 10;
            arri++;
            next = 1;
        }else{
            arri++;
            next = 0;
        }
    }

    if(next == 1){
        Arr[arri++] = 1;
    }
    *returnSize = arri;
    //Arr中的值与预期的值是相反的，需要逆置
    int left = 0,right = arri - 1;
    while(left < right){
        int t = Arr[left];
        Arr[left] = Arr[right];
        Arr[right] = t;
        ++left;
        --right;
    }

    return Arr;

}