codility Max-Nonoverlapping-Segments

Question:codility Lesson16 Max-Nonoverlapping-Segments

My Answer:

def solution(A,B):
    if len(A) < 2:
        return len(A)

    res = 1
    end = B[0]

    for i in range(1,len(A)):
        if A[i] > end:
            res += 1
            end = B[i]
    return res