[Programmers] 코딩 기초 트레이닝 181879. 길이에 따른 연산

✔️ Problem URL

길이에 따른 연산

✔️ Problem

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

✔️ Code

리팩토링 전

class Solution {
    public int solution(int[] num_list) {
        int length = 0, answer = 1;

        for(int i = 0; i < num_list.length; i++) {
            length++;
        }

        if(length >= 11) {
            for(int i : num_list) {
                answer += i;
            }
            return answer-1;
        }
        else {
            for(int i : num_list) {
                answer *= i;
            }
        }
        return answer;
    }
}

리팩토링 후

변수를 사용하지 않고 num_list.length를 이용하는 코드로 리팩토링 하니 for문도 훨씬 적게 쓰고 더 효율적이라는 것을 알 수 있었다. 역시 자바 문법은 복습이 답이다.

class Solution {
    public int solution(int[] num_list) {
        int answer = (num_list.length >= 11 ? 0 : 1);

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