Combine 2 sorted arrays

• Input: arr1: [1,2,3], arr2: [4,5,6]
• Output: [1,2,3,4,5,6]

Solution

def combine(arr1, arr2):
    i = 0
    j = 0

    output = []

    while i < len(arr1):
        if arr1[i] < arr2[j]:
            output.append(arr1[i])
            i += 1
        else:
            output.append(arr2[j])
            j += 1

    if j < len(arr2):
        output += arr2[j:]

    return output

combine([1,2,3], [4,5,6])

[1, 2, 3, 4, 5, 6]

combine([1, 5, 8], [2, 4, 9])

[1, 2, 4, 5, 8, 9]