blob: 6e19580d14852b2b165fdbdb8a9650e4b0f4e85c (
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
26
27
28
|
from itertools import combinations
def IsAriTriplet (arrInput, nDiffInput):
if arrInput[1] - arrInput[0] != nDiffInput:
return False
if arrInput[2] - arrInput[1] != nDiffInput:
return False
return True
## Example 1
## arrNum = [0, 1, 4, 6, 7, 10]
## nDiff = 3
## Example 2
arrNum = [4, 5, 6, 7, 8, 9]
nDiff = 2
arrCombList = combinations(arrNum, 3)
arrOutputList = []
for arrLoop in list(arrCombList):
if IsAriTriplet(arrLoop, nDiff):
arrOutputList.append(arrOutputList)
print (len(arrOutputList))
|
