원소들의 곱과 합
in Coding Test / Programmers
프로그래머스 파이썬 - 원소들의 곱과 합
원소들의 곱과 합 썸네일
문제
풀이
from functools import reduce
def multiply(x, y):
return x * y
def solution(num_list):
if sum(num_list) ** 2 > reduce(multiply, num_list):
return 1
else:
return 0
파이썬에서 제공 해주는 패키지인 reduce
를 사용해서 배열에 있는 정수들을 합할 수 있었다.
그리고 sum
을 사용해서 한 번에 계산한 후 **
제곱해주고 조건에 따라 리턴 해준다.