[TIL] 2026-01-19 | C++ Set 사용

알고리즘 코드카타

 

Set

#include <iostream>
#include <set>
using namespace std;

int main() {
	
	/* 생성자 */
	set<int> set;		// 오름차순, 기본 값 : less<int>
	//set<int, greater<int>> set;		//내림차순
	
	/* 삽입, 삭제 */
	set.insert(30);
	set.insert(40);
	set.insert(10);
	set.insert(20);
	set.insert(50);
	set.insert(50);		// 중복 저장 x
	set.erase(20);

	for (int i : set) {
		cout << i << " ";
	}
	cout << "\n";
	for (auto it = set.begin(); it != set.end(); it++) {
		cout << *it << " ";
	}
	cout << "\n";


	/* 값 탐색 */
	// 1. 삽입 성공여부 확인
	auto it = set.insert(50);	// pair< set<int>::iterator, bool >
	if (it.second) {
		cout << *it.first << "삽입 성공\n";
	}
	else {
		cout << *it.first << "삽입 실패\n";
	}


	// 2. 개수 탐색. set은 중복이 불가하므로 0 또는 1의 값
	cout << "원소 60 개수 : " << set.count(60) << "\n";


	// 3. find 함수로 탐색
	if (set.find(50) == set.end()) {	// find는 존재하지 않으면 set.end()를 리턴
		cout << "원소 50은 현재 없다.\n";
	}
	else {
		cout << "원소 50은 현재 존재한다.\n";
	}


	return 0;
}

 

 

프로그래머스 - 두 개 뽑아서 더하기

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

#include <string>
#include <vector>
#include <set>

using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    set<int> temp;
    for(int i=0;i<numbers.size()-1;i++){
        for(int j=i+1;j<numbers.size();j++){
            temp.insert(numbers[i]+numbers[j]);
        }
    }
    for(int i:temp)
        answer.push_back(i);
    return answer;
}

중복 X, 오름차순일 때는 Set을 사용해서 간단하게 구현할 수 있다.

참고

https://notepad96.tistory.com/25

[STL] Set 생성, 삽입, 삭제 등 사용법