Max Points

maxPoints.py

#!/bin/python3

#
# Complete the 'maxPoints' function below.
#
# The function is expected to return a LONG_INTEGER.
# The function accepts INTEGER_ARRAY elements as parameter.
#

def maxPoints(elements):
    possiblePoints = []
    possiblePoints.append(pointsSorting(elements[:], False, False))
    possiblePoints.append(pointsSorting(elements[:], True, False))
    possiblePoints.append(pointsSorting(elements[:], True, True))
    possiblePoints.append(pointsSortingLoop(elements[:], True, True))
    return max(possiblePoints)

def pointsSorting(elements, sort, reverse):
    points = 0
    if sort:
        elements.sort(reverse=reverse)
    while len(elements):
        maxElement = elements[0]
        points += sum([e for e in elements if e == maxElement])
        elements = [e for e in elements if e not in [maxElement, maxElement - 1, maxElement + 1]]
    return points

def pointsSortingLoop(elements, sort, reverse):
    points = 0
    if sort:
        elements.sort(reverse=reverse)
    previousMaxElement = elements[0]
    for element in elements:
        if element == previousMaxElement:
            points += element
        if element < previousMaxElement - 1:
            points += element
            previousMaxElement = element
    return points

if __name__ == '__main__':
    print(maxPoints([3, 4, 2])) # 6