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

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

프로그래머스

 

문제

코드 1

def solution(n) :
    if n%2 :
        return sum((range(1,n+1,2)))
    return sum([i**2 for i in range(2,n+1,2)])

코드 2

def solution(n):
    # n이 홀수인지 짝수인지 파악
    # 각 분류별로 n 이하의 홀수/짝수인 모든 정수의 합
    sum1 = 0
    sum2 = 0
    if n % 2 == 0 :
        for i in range(1, n+1) :
            if i % 2 == 0 :
                sum1 += i**2
        return sum1
    else :
        for j in range(1, n+1) :
            if j % 2 == 1 :
                sum2 += j
        return sum2