LeetCode 989.数组形式的整数加法

难度：简单。
考虑好各种情况就行。

class Solution {

public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        if(K == 0)return A;
        int A_num = A.size();
        
        int temp = 0;
        int a_index = A_num - 1;
        int m = K;
        for(int i = 0; i < 5; i++){
            int n = m % 10;
            int sum = n + temp;
            if(a_index >= 0){
                sum += A[a_index];
                if(sum > 9){
                    A[a_index] = sum % 10;
                    temp = 1;
                    a_index--;
                }
                else{
                    A[a_index] = sum;
                    a_index--;
                    temp = 0;
                }
            }
            else{
                if(sum > 9){
                    A.insert(A.begin(), sum % 10);
                    temp = 1;
                }
                else{
                    A.insert(A.begin(), sum);
                    temp = 0;
                }
            }
            
            
            m = m / 10;
            if(m == 0)break;
        }

        if(temp == 0)return A;
        for(int i = a_index; i >= 0; i--){
            int sum = A[i] + temp;
            if(sum > 9){
                A[i] = sum % 10;
                temp = 1;
            }
            else{
                A[i] = sum;
                temp = 0;
            }
        }
        if(temp == 1)A.insert(A.begin(), 1);
        return A;
    }
};