Day4 - 홀짝에 따라 다른 값 반환하기

프로그래머스 문제: https://school.programmers.co.kr/learn/courses/30/lessons/181935

프로그래머스

 

문제 풀이

def solution(n):
    answer = 0

    for i in range(n+1) :
        if(n % 2 == 0 and i%2==0) :
            answer = answer + i * i
            print(i)
        elif(n % 2 != 0 and i%2!=0) :
            answer = answer + i 
            print(i)
    return answer

 

1. 홀짝 여부 확인하기

2. 홀 - n이하 홀수 더하기, 짝 - n이하 짝수 더하기

이 두가지를 어떻게 하면 좋을지 고민 해보았습니다. 좋은 정답은 아닌 것 같습니다 ^