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

문제

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

 

솔루션

#include <string>
#include <vector>

using namespace std;

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

 

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

프로그래머스