[Coding Test]길이에 따른 연산(java, python,프로그래머스)

문제 설명

 

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

 

제한사항

2 ≤ num_list의 길이 ≤ 20
1 ≤ num_list의 원소 ≤ 9

입출력 예

 

입출력 예 #1
리스트의 길이가 13이므로 모든 원소의 합인 51을 return합니다.

 

입출력 예 #2
리스트의 길이가 4이므로 모든 원소의 곱인 120을 return합니다.

 

 

def solution(num_list):
    if len(num_list) >= 11:
        return sum(num_list)
    else:
        answer = 1
        for num in num_list:
            answer *= num
        return answer

 

 

class Solution {
    public int solution(int[] num_list) {
        int length = num_list.length;
        if (length >= 11) {
            int sum = 0;
            for (int num : num_list) {
                sum += num;
            }
            return sum;
        } else {
            int product = 1;
            for (int num : num_list) {
                product *= num;
            }
            return product;
            }
        }
    }