[python] - 홀짝에 따라 다른 값 반환하기

리스트 컴프리헨션

li = [x for x in range(1, n)]

내 코드

def solution(n):
    answer = 0
    list_n = [x for x in range(1, n+1)]

    if n % 2 == 0:
        even_list = list_n[1::2]
        for i in range(len(even_list)):
             answer += even_list[i] * even_list[i]
            
    else:
        odd_list = list_n[::2]
        for i in odd_list:
             answer += i
            
    return answer

더 좋은 코드 (프로그래머스)

def solution(n):
  answer = 0
  if n%2:
      for i in range(1,n+1,2):
          answer += i
  else:
      for i in range(2,n+1,2):
          answer += i**2
  return answer