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

문제 설명

정수가 담긴 리스트 num_list가 주어질 때, 리스트의 길이가 11 이상이면 리스트에 있는 모든 원소의 합을 10 이하이면 모든 원소의 곱을 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 2 ≤ num_list의 길이 ≤ 20
• 1 ≤ num_list의 원소 ≤ 9
• num_list의 원소를 모두 곱했을 때 2,147,483,647를 넘는 입력은 주어지지 않습니다.

---

입출력 예

num_list result

[3, 4, 5, 2, 5, 4, 6, 7, 3, 7, 2, 2, 1]	51
[2, 3, 4, 5]	120

---

입출력 예 설명

입출력 예 #1

• 리스트의 길이가 13이므로 모든 원소의 합인 51을 return합니다.

입출력 예 #2

• 리스트의 길이가 4이므로 모든 원소의 곱인 120을 return합니다.

#include <string>
#include <vector>

using namespace std;

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

음 그냥 for문으로 묶어서 할까 했지만 속도로 보면 이게 뭔가 더 빠를거같아서 이렇게 했다..ㅎㅎ..