728x90
문제 설명
양의 정수 n이 매개변수로 주어질 때,
n이 홀수라면 n 이하의 홀수인 모든 양의 정수의 합을 return 하고
n이 짝수라면 n 이하의 짝수인 모든 양의 정수의 제곱의 합을 return 하는 solution 함수를 작성해 주세요.
제한 사항
1 ≤ n ≤ 100
예제
# 1
n = 7
return 16
# 2
n = 10
return 220문제 풀이
def solution(n):
# 짝수/홀수 판별
if n % 2 != 0:
# 홀수일 경우 홀수에 해당하는 숫자만 answer로 선언
answer = [i for i in range(1,n+1,2)]
else:
# 짝수일 경우 짝수에 해당하는 숫자를 제곱하여 answer로 선언
answer = [i**2 for i in range(2,n+1,2)]
return sum(answer)파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.
728x90
