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

提示：

1. 1 <= A.length <= 10000
2. 0 <= A[i] <= 9
3. 0 <= K <= 10000
4. 如果 A.length > 1，那么 A[0] != 0

---

class Solution {
public:
	vector<int> addToArrayForm(vector<int>& A, int K) {
		vector<int> tmp, tk, ans, res;
		while (K != 0) {
			tmp.push_back(K % 10);
			K /= 10;
		}
		for (int i = tmp.size() - 1; i >= 0; i--) {
			tk.push_back(tmp[i]);
		}
		int i = tk.size() - 1, j = A.size() - 1;
		int cnt = 0;
		for (; i >= 0 || j >= 0; i--, j--) {
			if (i >= 0 && j >= 0) {
				if (tk[i] + A[j] + cnt < 10) {
					ans.push_back(tk[i] + A[j] + cnt);
					cnt = 0;
				}
				else {
					ans.push_back((tk[i] + A[j] + cnt) % 10);
					cnt = 1;
				}
			}
			else if (i < 0 && j >= 0) {
				if (A[j] + cnt < 10) {
					ans.push_back(A[j] + cnt);
					cnt = 0;
				}
				else {
					ans.push_back((A[j] + cnt) % 10);
					cnt = 1;
				}
			}
			else if (j < 0 && i >= 0) {
				if (tk[i] + cnt < 10) {
					ans.push_back(tk[i] + cnt);
					cnt = 0;
				}
				else {
					ans.push_back((tk[i] + cnt) % 10);
					cnt = 1;
				}
			}
		}
		if (cnt == 1) ans.push_back(1);
		for (int i = ans.size() - 1; i >= 0; i--) {
			res.push_back(ans[i]);
		}
		return res;
	}
};