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

문제

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

- 제한사항

• 2 ≤ num_list의 길이 ≤ 20
• 1 ≤ num_list의 원소 ≤ 9
• num_list의 원소를 모두 곱했을 때 2,147,483,647를 넘는 입력은 주어지지 않습니다.

풀이

리스트의 길이가 10 이하면 모든 원소를 곱해야 하기 때문에 int answer = 1;로 초기화하고, 길이가 11 이상일 때 올바른 값을 리턴하기 위해 answer -= 1; 해준다.

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

결과