프로그래머스/0단계
원소들의 곱과 합
긴가우딘
2024. 6. 10. 02:04
문제 설명
정수가 담긴 리스트 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;
}
}