[프로그래머스/Java] Lv.0 - 길이에 따른 연산

문제

https://school.programmers.co.kr/learn/courses/30/lessons/181879

프로그래머스

 

문제 설명

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

 

풀이

배열의 길이가 11이상인 경우 배열을 순회하며 원소의 합을 answer에 누적한다. 미만인경우 곱하기를 수행하므로 초기 answer의 값을 num_list[0]의 값으로 지정하고 이후 원소의 곱을 answer에 누적한다.

 

코드

class Solution {
    public int solution(int[] num_list) {
        int answer = 0;
        if(num_list.length >= 11) {
            for (int i = 0 ; i < num_list.length ; i++) {
                answer += num_list[i];    
            }
        } else {
            for (int i = 0; i < num_list.length ; i++) {
                if (answer == 0) answer = num_list[i];
                else            answer *= num_list[i];
            }
        }
        return answer;
    }
}