[자바]길이에 따른 연산

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

프로그래머스

 

문제 설명

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

 

제한사항

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

입출력 예 설명

입출력 예 #1

• 리스트의 길이가 13이므로 모든 원소의 합인 51을 return합니다.

입출력 예 #2

• 리스트의 길이가 4이므로 모든 원소의 곱인 120을 return합니다.

나의 문제 풀이

1) 배열의 길이를 먼저 파악한다.

2) 길이가 11 >= 이면 모두 더한다.

3) 길이가 11 < 이면 곱해야 하는데 answer = 0 이여서 처음 결과 값이 0이 나왔다.

4) 그래서 for문이 시작되기 전에 answer = 1로 다시 지정하고 for문을 실행 하였다.

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