descripción
Para el entero no negativo X, la forma de matriz de X es una matriz formada por cada dígito de izquierda a derecha. Por ejemplo, si X = 1231, entonces su forma de matriz es [1,2,3,1].
Dada la forma de matriz A del entero no negativo X, devuelve la forma de matriz del entero X + K.
Ejemplo 1:
Entrada: A = [1,2,0,0], K = 34
Salida: [1,2,3,4]
Explicación: 1200 + 34 = 1234
Ejemplo 2:
Entrada: A = [2,7,4], K = 181
Salida: [4,5,5]
Explicación: 274 + 181 = 455
Ejemplo 3:
Entrada: A = [2,1,5], K = 806
Salida: [1,0,2,1]
Explicación: 215 + 806 = 1021
Ejemplo 4:
Entrada: A = [9,9,9,9,9,9,9,9,9,9], K = 1
Salida: [1,0,0,0,0,0,0,0,0, 0,0]
Explicación: 9999999999 + 1 = 10000000000
rápido:
1 <= A. longitud <= 10000
0 <= A [i] <= 9
0 <= K <= 10000
如果 A. longitud> 1 , 那么 A [0]! = 0
Fuente: LeetCode
Enlace: https://leetcode-cn.com/problems/add-to-array-form-of-integer/
Resolver
class Solution {
public:
vector<int> addToArrayForm(vector<int> &A, int K) {
const int n = A.size();
vector<int> res;
res.reserve(n + 1);
int carry = 0; // 进位
constexpr int DECIMAL = 10; // 十进制常量
auto riter = A.crbegin();
auto rend = A.crend();
// 注意循环条件,只有有一个没有被加数没有被使用完成,就应该继续相加
while (riter != rend || K != 0) {
int a = riter != rend ? *riter++ : 0; // 如果A循环没有完成取当前位数,如果完成取0
int sum = K % DECIMAL + a + carry;
K = K / DECIMAL;
res.emplace_back(sum % DECIMAL);
carry = sum / DECIMAL;
}
// 累加完成后需要判断最后是否有进位
if (carry > 0) {
res.emplace_back(carry);
}
return vector<int>(res.crbegin(), res.crend());
}
};