InterviewBit/Array Sum.cpp at main · kunalbandooni/InterviewBit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
vector<int> Solution::addArrays(vector<int> &A, vector<int> &B) {
    int a = A.size() - 1, b = B.size() - 1, carry = 0, sum = 0, last = 0;
    vector<int> ans;
    while(a >= 0 && b >= 0) {
        sum = 0;
        sum += carry + A[a] + B[b];
        if(sum > 9) {
            ans.push_back(sum % 10);
            sum = sum / 10;
            carry = sum;
        }
        else {
            ans.push_back(sum);
            carry = 0;
        }
        a--;
        b--;
    }
    while(a >= 0) {
        sum = 0;
        sum += carry + A[a];
        if(sum > 9) {
            ans.push_back(sum % 10);
            sum = sum / 10;
            carry = sum;
        }
        else {
            ans.push_back(sum);
            carry = 0;
        }
        a--;
    }
    while(b >= 0) {
        sum = 0;
        sum += carry + B[b];
        if(sum > 9) {
            ans.push_back(sum % 10);
            sum = sum / 10;
            carry = sum;
        }
        else {
            ans.push_back(sum);
            carry = 0;
        }
        b--;
    }
    if(carry > 0)
        ans.push_back(carry);
    reverse(ans.begin(), ans.end());
    return ans;
}