[프로그래머스] Lv.0 - 원소들의 곱과 합.py

문제

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

제한사항

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

입출력 예시

• 입출력 예 1
  • 모든 원소의 곱은 120, 합의 제곱은 225이므로 1을 return합니다.
• 입출력 예2
  • 모든 원소의 곱은 840, 합의 제곱은 529이므로 0을 return합니다.

코드 1

def solution(num_list):
    answer = 0
    sum_squre = 0
    mul = 1
    sum_all = 0

    for item in num_list:
        mul = item * mul
        sum_all = item + sum_all
    sum_squre = sum_all * sum_all

    if mul < sum_squre:
        answer = 1
    elif mul > sum_squre:
        answer = 0

    return answer

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