[프로그래머스] 원소들의 곱과 합

[unrated] 원소들의 곱과 합 - 181929

문제 링크

성능 요약

메모리: 10.1 MB, 시간: 0.00 ms

구분

코딩테스트 연습 > 코딩 기초 트레이닝

채점결과

Empty

문제 설명

정수가 담긴 리스트 num_list가 주어질 때, 모든 원소들의 곱이 모든 원소들의 합의 제곱보다 작으면 1을 크면 0을 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 2 ≤ num_list의 길이 ≤ 10
• 1 ≤ num_list의 원소 ≤ 9

---

입출력 예

num_list	result
[3, 4, 5, 2, 1]	1
[5, 7, 8, 3]	0

---

입출력 예 설명

입출력 예 #1

• 모든 원소의 곱은 120, 합의 제곱은 225이므로 1을 return합니다.

입출력 예 #2

• 모든 원소의 곱은 840, 합의 제곱은 529이므로 0을 return합니다.

정답

def solution(num_list):
    answer = 0
    multiply = 1
    add = 0
    
    for i in num_list:
        multiply *= i
        add += i
        
    answer = 1 if multiply < add*add else 0
    
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges