[오늘의 문제] 길이에 따른 연산

shlim55·2025년 12월 14일

코딩테스트

목록 보기
199/223

출처: https://school.programmers.co.kr/learn/courses/30/lessons/181879?language=python3

문제 설명
정수가 담긴 리스트 num_list가 주어질 때, 리스트의 길이가 11 이상이면 리스트에 있는 모든 원소의 합을 10 이하이면 모든 원소의 곱을 return하도록 solution 함수를 완성해주세요.

제한사항
2 ≤ num_list의 길이 ≤ 20
1 ≤ num_list의 원소 ≤ 9
num_list의 원소를 모두 곱했을 때 2,147,483,647를 넘는 입력은 주어지지 않습니다.
입출력 예
num_list result
[3, 4, 5, 2, 5, 4, 6, 7, 3, 7, 2, 2, 1] 51
[2, 3, 4, 5] 120

내가 한 풀이

def solution(num_list):
    answer = 0
    
    if len(num_list) >= 11:# 리스트에 있는 모든 원소의 합을
        for i in range(len(num_list)):
            answer+=num_list[i]
    else:   
        answer = 1
        for i in range(len(num_list)):
            answer*=num_list[i]
    
    return answer
profile
A Normal Programmer

0개의 댓글