C++ 코딩테스트 주요 문법 정리
프로그래머스 PCCP Lv.3 이상을 목표로 코딩테스트 준비를 위해 C++ 코딩테스트에서 주로 사용되는 문법 등을 정리한다. 1. 기본 (입출력 최적화) int main() { ios::sync_with_stdio(false); cin.tie(NULL); } ios::sync_with_stdio(false) : C / C++ 입출력 동기화 해제 cin.tie(NULL) : cout flush 방지 대량 입출력 문제에서 시간초과 방지 2. STL 컨테이너 2.1 vector vector<int> v; v.push_back(10); v.push_back(20); cout << v[0]; v.pop_back(); sort 정렬 sort(v.begin(), v.end()); // 오름차순 sort(v.begin(), v.end(), greater<int>()); // 내림차순 커스텀 정렬 sort(v.begin(), v.end(), [](int a, int b) { return a > b; }); erase-remove 특정 값 삭제 vector<int> v = {1,3,2,3,4}; v.erase(remove(v.begin(), v.end(), 3), v.end()); <blockquote> <blockquote> 결과 1 2 4 unique 중복 제거 sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); lower_bound / upper_bound vector<int> v = {1,2,2,2,3}; int idx = lower_bound(v.begin(), v.end(), 2) - v.begin(); <blockquote> <blockquote> 결과 1 lower_bound : x 이상 첫 위치 upper_bound : x 초과 첫 위치 binary_search vector<int> v = {1,2,3,4,5}; bool exist = binary_search(v.begin(), v.end(), 3); 2.2 string find 문자열 검색 string s = "hello world"; int pos = s.find("world"); if(pos == string::npos) { cout << "not found"; } substr 문자열 자르기 string s = "abcdef"; cout << s.substr(2,3); <blockquote> <blockquote> 결과 cde replace 문자열 치환 string s = "abcdef"; s.replace(2,3,"ZZ"); <blockquote> <blockquote> 결과 abZZf 여러 번 치환 string s = "ayaaya"; int pos; while((pos = s.find("aya")) != string::npos) { s.replace(pos,3,"!"); } erase 문자 삭제 string s = "abcdef"; s.erase(2,3); <blockquote> <blockquote> 결과 abf getline 공백 포함 입력 string line; while(getline(cin,line)) { cout << line; } 2.3 pair 두 값을 묶어서 저장 pair<int,int> p = {3,5}; cout << p.first; cout << p.second; BFS에서 많이 사용 queue<pair<int,int>> q; 2.4 map 정렬된 key-value 저장 map<string,int> m; m["apple"] = 3; cout << m["apple"]; 자동 정렬 시간복잡도 O(logN) 2.5 unordered_map 해시맵 unordered_map<string,int> m; m["apple"] = 3; 평균 O(1) 정렬 없음 2.6 set 중복 없는 집합 set<int> s; s.insert(3); s.insert(3); s.insert(5); for(int x : s) { cout << x; } 출력 3 5 2.7 stack stack<int> s; s.push(10); s.push(20); cout << s.top(); s.pop(); 2.8 queue queue<int> q; q.push(10); q.push(20); cout << q.front(); q.pop(); 2.9 priority_queue (힙) priority_queue<int> pq; pq.push(5); pq.push(10); pq.push(1); cout << pq.top(); 결과 10 최소 힙 priority_queue<int, vector<int>, greater<int>> pq; 3. 그래프 탐색 BFS queue<int> q; q.push(1); while(!q.empty()) { int cur = q.front(); q.pop(); for(int next : graph[cur]) { if(!visited[next]) { visited[next] = true; q.push(next); } } } DFS void dfs(int node) { visited[node] = true; for(int next : graph[node]) { if(!visited[next]) { dfs(next); } } } 격자 탐색 int dx[4] = {1,-1,0,0}; int dy[4] = {0,0,1,-1}; for(int i=0;i<4;i++) { int nx = x + dx[i]; int ny = y + dy[i]; } 4. 알고리즘 패턴 Prefix Sum (누적합) vector<int> psum(n+1); for(int i=1;i<=n;i++) { psum[i] = psum[i-1] + arr[i]; } 구간합 sum = psum[r] - psum[l-1]; Two Pointer int l = 0; int r = 0; while(r < n) { sum += arr[r]; while(sum > target) { sum -= arr[l]; l++; } r++; } Binary Search int l = 0; int r = n-1; while(l <= r) { int mid = (l+r)/2; if(arr[mid] == target) break; if(arr[mid] < target) l = mid + 1; else r = mid - 1; } 5. math #include <cmath> 함수 설명 abs(x) 절댓값 sqrt(x) 루트 pow(a,b) 거듭제곱 ceil(x) 올림 floor(x) 내림
