[프로그래머스] 홀짝에 따라 다른 값 반환하기

 

<내가 쓴 코드>

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