1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | struct Data { int a; int b; bool operator>(const Data& data) { return this->b > data.b; } }; struct Heap { Data data[300000]; int k = 1; void init() { k = 1; } bool empty() { if (k == 1) return true; return false; } void push(Data input) { int cur = k++; data[cur] = input; int par = cur / 2; while (1) { if (par == 0) break; if (data[par] > data[cur]) { swap(data[par], data[cur]); cur = par; par = cur / 2; } else break; } } Data pop() { Data result = data[1]; swap(data[1], data[k - 1]); k--; int cur = 1; while (1) { int child = cur * 2; if (child >= k) break;//자식이 없다면 종료 if (child + 1 < k) {//오른쪽 자식이 존재한다면 if (data[child] > data[child + 1]) child += 1; } if (data[cur] > data[child].) { swap(data[cur], data[child]); cur = child; } else break; } return result; } }; | cs |
'알고리즘(Algorithm) > 삼성 B형 준비' 카테고리의 다른 글
이분탐색 (0) | 2019.12.22 |
---|