aboutsummaryrefslogtreecommitdiff
path: root/challenge-200/eric-cheung/python/ch-1.py
blob: 74f509b8213bb6fe6045c914bcd4290146f34f58 (plain)
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
## arrInput = [1, 2, 3, 4]  ## Example 1
arrInput = [2]  ## Example 2

def IsArithmetic(arrSubInput):

    if len(arrSubInput) < 3:
        return False

    nDiff = arrSubInput[1] - arrSubInput[0]

    for nLoop in range(2, len(arrSubInput)):
        if arrSubInput[nLoop] - arrSubInput[nLoop - 1] != nDiff:
            return False

    return True

arrOutput = []

for nLen in range(3, len(arrInput) + 1):
    for nIndx in range(0, len(arrInput) - nLen + 1):
        if IsArithmetic(arrInput[nIndx:nIndx + nLen]):
            arrOutput.append(arrInput[nIndx:nIndx + nLen])

print (arrOutput)