[C++/PGS] Lv.0 : 연속된 수의 합 (구현)

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

 

문제 설명

연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num과 total이 주어집니다. 연속된 수 num개를 더한 값이 total이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.

---

제한사항

• 1 ≤ num ≤ 100
• 0 ≤ total ≤ 1000
• num개의 연속된 수를 더하여 total이 될 수 없는 테스트 케이스는 없습니다.

---

코딩테스트 입문 문제 중에

정답률이 낮은 문제들 중 하나이다.

테스트케이스에서 규칙을 찾아서 공식화시켜야 한다!

 

이 예시를 보니

result의 가운데 숫자는 항상 total/num의 값(mid)이다.

그러면 num이 홀수일 때 리스트의 시작값은 mid - num/2이다.

그런데 num이 짝수일 때는 num/2번째에 mid가 있으므로 시작값이 mid - num/2 + 1이 된다.

 

 

 

나의 풀이

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int num, int total) {
    vector<int> answer;
    int mid = total/num;
    int start = mid - num/2;
    if(num%2==0) start += 1;
    for(int i=0; i<num; ++i){
        answer.push_back(start+i);
    }
    return answer;
}