C++ 두 개 뽑아서 더하기 - Programmers

코딩테스트 연습 두 개 뽑아서 더하기

---

두 개 뽑아서 더하기

문제 설명

정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

제한 사항

• numbers의 길이는 2 이상 100 이하입니다.
  • numbers의 모든 수는 0 이상 100 이하입니다.

입출력 예

numbers = [2,1,3,4,1]
result = [2,3,4,5,6,7]

입출력 예 설명

입출력 예 #1

• 2 = 1 + 1 입니다. (1이 numbers에 두 개 있습니다.)
• 3 = 2 + 1 입니다.
• 4 = 1 + 3 입니다.
• 5 = 1 + 4 = 2 + 3 입니다.
• 6 = 2 + 4 입니다.
• 7 = 3 + 4 입니다.
• 따라서 [2,3,4,5,6,7] 을 return 해야 합니다.

입출력 예 #2

• 2 = 0 + 2 입니다.
• 5 = 5 + 0 입니다.
• 7 = 0 + 7 = 5 + 2 입니다.
• 9 = 2 + 7 입니다.
• 12 = 5 + 7 입니다.
• 따라서 [2,5,7,9,12] 를 return 해야 합니다.

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> numbers) {  
  vector<int> answer;  
  vector<int>::iterator it = answer.begin();  
	int temp = 0;  
	int size = numbers.size();  

  for (int i = 0; i < size - 1; i++)  
	{  
		for (int j = i + 1; j < size; j++)  
		{  
			temp = numbers[i] + numbers[j];

			it = find(answer.begin(), answer.end(), temp); // 중복된 값이 있는지 확인
			if (it == answer.end())  // 중복된 값 없으면 넣기
			{
				answer.push_back(temp);
			}
		}
	}

    sort(answer.begin(), answer.end());
    return answer;
}

고찰

vector만 사용하면서 풀려고 하니 불필요한 줄이 늘어나는 것 같다. stack, queue, set 등 여러 가지들을 공부해서 쉽게 푸는 것을 연습해봐야겠다. 이번 문제같은 경우 set을 사용하면 자동으로 sorting을 해주고 중복된 값은 넣지 않으므로 set으로 하면 훨씬 간단하게 코딩할 수 있을 것이다.