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

<내가 푼 풀이>

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