[Python] 프로그래머스 : 길이에 따른 연산 181879, sum & prod

 

 

📝 문제

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

 

🫠 나의 풀이

from functools import reduce

def solution(num_list):
    if len(num_list) >= 11:
        return sum(num_list)
    else:
        return reduce(lambda x, y: x*y, num_list)

 

🧞‍♂️ 다른 사람의 풀이

from math import prod

def solution(num_list):
    return sum(num_list) if len(num_list)>=11 else prod(num_list)

 

sum 처럼 한 번에 해결할 수 있는 메서드가 있을 것 같아서 찾다가 lambda 를 사용했는데

prod 라는게 있었다! 😳

 

 

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

프로그래머스