본문 바로가기
프로그래머스/0단계

원소들의 곱과 합

by 긴가우딘 2024. 6. 10.

문제 설명

정수가 담긴 리스트 num_list가 주어질 때, 모든 원소들의 곱이 모든 원소들의 합의 제곱보다 작으면 1을 크면 0을 return하도록 solution 함수를 완성해주세요.


제한사항

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

입출력 예

num_list result

[3, 4, 5, 2, 1] 1
[5, 7, 8, 3] 0

입출력 예 설명

입출력 예 #1

  • 모든 원소의 곱은 120, 합의 제곱은 225이므로 1을 return합니다.

입출력 예 #2

  • 모든 원소의 곱은 840, 합의 제곱은 529이므로 0을 return합니다.

 

class Solution {
    public int solution(int[] num_list) {
        int product = 1;
        int sum = 0;
        int squareSum = 1;
        
        for(int i=0; i<num_list.length; i++) {
            
            product *= num_list[i];
            sum += num_list[i];
            squareSum = sum * sum;
        }
        
        if(product < squareSum) {
                return 1;
            }   
        else 
            return 0;
    }
}

 

foreach와 삼항연산자를 사용하는 연습을 해보자

 

class Solution {
    public int solution(int[] num_list) {
        int answer = 0;
        int product = 1;
        int sum = 0;
        
        for(int i : num_list) {
            product *= i;
            sum += i;
        }
        
        answer = (product < sum*sum) ? 1 : 0;
        
        return answer;
    }
}

 

'프로그래머스 > 0단계' 카테고리의 다른 글

[PCCE 기출문제] 7번 / 버스  (1) 2025.01.06
[PCCE 기출문제] 6번 / 물 부족  (0) 2025.01.06
주사위 게임2  (0) 2024.06.06
문자열 섞기  (0) 2024.05.21
문자열 겹쳐쓰기  (0) 2024.05.18