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

문제

https://school.programmers.co.kr/learn/courses/30/lessons/181935

프로그래머스

 

Code

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