알고리즘 - 홀짝에 따라 다른 값 반환하기

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