STL을 사용하여 문제 풀이 연습하기

[capture](parameters) -> return_type { // 함수 본문 }

auto add = [](int a, int b) -> int { return a + b; }; int result = add(3, 4); // result는 7

int x = 10; int y = 20; auto addXY = [x, &y](int a) { return a + x + y; }; int result = addXY(5); // result는 35, x는 10, y는 20

int a = 1, b = 1; auto fib = [=]() mutable { int temp = a; a = b; b = temp + b; return temp; }; for (int i = 0; i < 5; ++i) { std::cout << fib() << " "; // 1 1 2 3 5 }

#include <algorithm> #include <vector> #include <iostream> int main() { std::vector<int> vec = { 1, 2, 3, 4, 5 }; std::for_each(vec.begin(), vec.end(), [](int& n) { n *= 2; }); for (int n : vec) { std::cout << n << " "; // 2 4 6 8 10 } return 0; }

template<class RandomIt> void sort(RandomIt first, RandomIt last); template<class RandomIt, class Compare> void sort(RandomIt first, RandomIt last, Compare comp);

#include <bits/stdc++.h> #define fastio cin.tie(0)->sync_with_stdio(0) using namespace std; // 구조체 내부에 비교 연산자를 정의 struct Info { int val; // operator < 를 오버로딩하여 정렬 기준을 정의 // sort 함수는 기본적으로 < 연산자를 사용하므로 // 이 연산자를 재정의하면 정렬 순서를 변경할 수 있음 bool operator < (const Info& i) const { return val > i.val; // val이 큰 원소가 앞에 오도록 정렬 (내림차순) } }; int main() { fastio; vector<Info> v; // 1부터 5까지의 값을 가진 Info 객체를 벡터에 추가 for (int i = 1; i <= 5; i++) v.push_back(Info{ i }); // sort 함수 호출 시 Info 구조체의 operator < 가 자동으로 사용됨 sort(v.begin(), v.end()); // 결과 출력: 5 4 3 2 1 (내림차순) for (auto& i : v) cout << i.val << ' '; cout << '\n'; }

#include <bits/stdc++.h> #define fastio cin.tie(0)->sync_with_stdio(0) using namespace std; // 전역 비교 함수 정의 // 문자열의 길이를 우선 비교하고, 길이가 같으면 사전 역순으로 정렬 bool Cmp(const string& a, const string& b) { // 1순위: 문자열 길이 비교 (짧은 것이 앞으로) if (a.size() != b.size()) return a.size() < b.size(); // 2순위: 길이가 같은 경우 사전 역순 정렬 (z에 가까운 것이 앞으로) return a > b; } int main() { fastio; vector<string> v{ "stack", "queue", "list", "set", "map" }; // sort 함수의 세 번째 인자로 비교 함수를 전달 sort(v.begin(), v.end(), Cmp); // 결과: set map(길이 3) -> list(길이 4) -> stack queue(길이 5, 사전 역순) for (auto& i : v) cout << i << ' '; cout << '\n'; }

auto compare = [](int a, int b) { return a < b; }

#include <iostream> #include <cstdio> #include <vector> #include <algorithm> using namespace std; int main() { vector<int> numbers = {5, 4, 3, 2, 1}; // 비교 함수 내부에서 비교 과정을 출력 sort(numbers.begin(), numbers.end(), [](int a, int b) { // a와 b가 어떤 순서로 비교되는지 출력 printf("비교: a=%d, b=%d -> ", a, b); bool result = a < b; printf("결과=%s\n", result ? "true(a를 앞에)" : "false(순서 유지)"); return result; }); cout << "\nSorted numbers in ascending order: "; for (const auto& num : numbers) { cout << num << " "; } cout << endl; // 실행 결과를 보면: // - a에는 현재 삽입하려는 요소가 들어옴 // - b에는 이미 정렬된 부분의 요소가 들어옴 // - a < b가 true면 a를 b보다 앞에 배치 → 작은 수가 앞으로 → 오름차순 return 0; }

#include <iostream> #include <vector> #include <algorithm> using namespace std; int main() { // 정렬할 정수형 벡터 초기화 vector<int> numbers = {5, 2, 8, 1, 7, 3}; // sort 함수를 사용한 오름차순 정렬 // 비교 함수를 생략하면 기본적으로 < 연산자를 사용 (오름차순) sort(numbers.begin(), numbers.end()); cout << "Sorted numbers in ascending order: "; for (const auto& num : numbers) { cout << num << " "; } cout << endl; return 0; }

#include <iostream> #include <vector> #include <algorithm> using namespace std; int main() { vector<int> numbers = {5, 2, 8, 1, 7, 3}; // 람다식으로 내림차순 비교 함수 정의 // a > b가 true면 "큰 값(a)을 앞에 배치" → 내림차순 sort(numbers.begin(), numbers.end(), [](int a, int b) { return a > b; }); cout << "Sorted numbers in descending order: "; for (const auto& num : numbers) { cout << num << " "; } cout << endl; // 출력 결과: 8 7 5 3 2 1 return 0; }

#include <iostream> #include <vector> #include <algorithm> using namespace std; // 사람 정보를 담는 구조체 struct Person { string name; int age; }; int main() { vector<Person> people = { {"Alice", 25}, {"Bob", 30}, {"Charlie", 22}, {"David", 28} }; // 나이를 기준으로 오름차순 정렬 // const Person&로 참조를 받아 불필요한 복사를 방지 // -> bool은 반환 타입을 명시 (생략 가능) sort(people.begin(), people.end(), [](const Person& a, const Person& b) -> bool { return a.age < b.age; // 나이가 적은 사람이 앞에 오도록 }); cout << "Sorted people by age:" << endl; for (const auto& person : people) { cout << person.name << " (" << person.age << ")" << endl; } // 출력: // Charlie (22) // Alice (25) // David (28) // Bob (30) return 0; }

#include <iostream> #include <vector> #include <algorithm> #include <numeric> // iota 함수를 위해 필요 using namespace std; int main() { // 원본 배열: 인덱스 0=5, 1=2, 2=8, 3=1, 4=7, 5=3 vector<int> numbers = {5, 2, 8, 1, 7, 3}; // 인덱스를 저장할 벡터 (원본 배열과 같은 크기) vector<int> indices(numbers.size()); // iota: 0부터 시작하여 1씩 증가하는 값으로 초기화 // indices = {0, 1, 2, 3, 4, 5} iota(indices.begin(), indices.end(), 0); // 핵심: 인덱스 배열을 정렬하되, 비교는 원본 배열의 값을 기준으로 // [&numbers]: 람다식에서 numbers 배열을 참조로 캡처 sort(indices.begin(), indices.end(), [&numbers](int i1, int i2) { // i1, i2는 인덱스 값 // numbers[i1], numbers[i2]는 해당 인덱스의 실제 값 return numbers[i1] < numbers[i2]; // 원본 배열 값으로 비교 }); // 정렬된 인덱스 순서 출력 cout << "Sorted indices based on the values: "; for (int i : indices) { cout << i << " "; // 예상 출력: 3 1 5 0 4 2 } cout << endl; // 해석: numbers[3]=1(가장 작음) -> numbers[1]=2 -> ... -> numbers[2]=8(가장 큼) // 인덱스를 사용하여 정렬된 순서로 원본 값 출력 cout << "Original array in sorted order: "; for (int i : indices) { cout << numbers[i] << " "; // 출력: 1 2 3 5 7 8 } cout << endl; // 원본 배열은 그대로 유지됨 cout << "Original array unchanged: "; for (int n : numbers) { cout << n << " "; // 출력: 5 2 8 1 7 3 } cout << endl; return 0; }