본문 바로가기
코딩테스트/자바 Level 0

[Java] 길이에 따른 연산

by onggury 2023. 7. 25.

문제

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

 

 

제한사항

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

 

import java.util.Arrays;

class Solution {
    public int solution(int[] num_list) {
        int answer = 1;
        if(num_list.length < 11) {
            for(int num : num_list) {
                answer *= num;
            }
        }
        else {
            answer = Arrays.stream(num_list).sum();
        }
        return answer;
    }
}

문제 조건대로 나누어 진행하였고

덧셈은 stream을 사용하였다.

 

 

출처

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