💻 자바 알고리즘 - 길이에 따른 연산

📕 문제 설명

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

 

📕 제한 사항

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

 

📝 답안

class Solution {
    public int solution(int[] num_list) {
        int answer = 0;
        boolean flag = (num_list.length>10)? true : false;

        for(int i=0; i<num_list.length; i++) {
            if(flag) {
                answer+= num_list[i];
            } else {
                answer=(answer==0)?num_list[i]:answer*num_list[i];
            }
        }
        return answer;
    }
}

num_list 배열의 길이가 10보다 크면 true, 아니면 false를 담는 boolean 타입의 flag를 통해 for문을 돌면서 분기처리를 한다. 이 때 else문으로 빠졌을 경우 answer가 0이라면 num_list의 원소를 담고, 아니라면 answer와 원소를 곱한 값을 담는다. 0과 원소를 곱하면 0이 되기 때문에 이와 같은 처리가 필요하다.

이러한 처리가 복잡하다면 answer 초기값을 0으로 두는 것이 아닌 num_list의 길이에 따라 0 또는 1로 해준다.

 

💡 참고

• 배열.length : 배열의 길이를 리턴