刷题-力扣-989

989. 数组形式的整数加法

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-to-array-form-of-integer/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

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

题目分析

  1. 整数相加,大于10就进位

代码

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        vector<int> b;
        vector<int> res;
        while (K > 0) {
            b.push_back(K % 10);
            K = K / 10;
        }
        int now = 0;
        if (A.size() > b.size()) {
            int j = A.size() - 1;
            for (int i = 0; i < b.size(); i++) {
                now = now >= 10 ? A[j] + b[i] + 1 : A[j] + b[i];
                b[i] = now >= 10 ? now - 10 : now;
                j--;
            }
            while (j >= 0) {
                now = now >= 10 ? A[j] + 1 : A[j];
                b.push_back(now >= 10 ? now - 10 : now);
                j--;
            }
            if (now >= 10) b.push_back(1);
        } else {
            int i = 0;
            for (int j = A.size() - 1; j >= 0; j--) {
                now = now >= 10 ? A[j] + b[i] + 1 : A[j] + b[i];
                b[i] = now >= 10 ? now - 10 : now;
                i++;
            }
            while (i < b.size()) {
                now = now >= 10 ? b[i] + 1 : b[i];
                b[i] = now >= 10 ? now - 10 : now;
                i++;
            }
            if(now >= 10) b.push_back(1);
        }
        for (int k = (b.size() - 1); k >= 0; k--) {
            res.push_back(b[k]);
        }
        return res;
    }
};

用时45min

posted @ 2021-01-22 20:39  韩亚光  阅读(65)  评论(0编辑  收藏  举报