input: Write a function to find the maximum element in the given array. If the array is empty, return None. If the array contains only one element, return that element.
  def find_max_element(array):
      if len(array) == 0:
          return None
      if len(array) == 1:
          return array[0]
      return max(array)
  
  assert find_max_element([1, 10, 50, 3, 8]) == 50
  assert find_max_element([-1, -5, -10, -3, -8]) == -1
  assert find_max_element([42]) == 42
  assert find_max_element([]) == None
output: def find_max_element(array):
      if len(array) == 0:
          return None
      if len(array) == 1:
          return array[0]
      return max(array)