[프로그래머스 / C++] 길이에 따른 연산

문제

길이에 따른 연산 : 문제 링크

문제 분석

• 정수가 담긴 리스트 num_list가 주어질 때, 리스트의 길이가 11 이상이면 리스트에 있는 모든 원소의 합을 10 이하이면 모든 원소의 곱을 return
• 리스트의 길이가 10이하인 경우 모든 원소의 곱을 return 해야 하므로, 마지막에 return할 answer은 1로 초기화.
• 리스트의 길이가 11 이상이면, for loop에서 각 원소의 합을 answer에 저장한 후, 초기값을 1로 초기화 했으므로 마지막에 1을 빼줌. 10 이하의 경우, for loop에서 각 원소의 곱을 answer에 저장. 최종적으로 저장된 answer를 return

풀이

#include <vector>

using namespace std;

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