[C++] 프로그래머스 원소들의 곱과 합

문제설명

정수가 담긴 리스트 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

풀이

모든 원소들의 곱과 합을 구하고 비교하는 로직을 구현하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> num_list) {
    int answer = 0;
    int mul = 1;
    int sum = 0;
    int sum_mul = 0;
    
    for(int i = 0; i < num_list.size(); i++)
    {
        mul *= num_list[i];
        sum += num_list[i];
    }
    
    sum_mul = sum * sum;
    
    if(mul < sum_mul)
        answer = 1;
    
    return answer;
}

결과