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
| void Heap::erase() { if (vec.empty()) return; int s = int(vec.size()); swap(vec[0], vec[s - 1]); vec.pop_back(); s -= 1; int i = 0; while (1) { if ((2 * i + 1) <= (s - 1)) { if ((2 * i + 2) <= s - 1) { if (vec[i] < vec[2 * i + 1] && vec[i] < vec[2 * i + 2]) { int temp = vec[2 * i + 1] < vec[2 * i + 2] ? (2 * i + 2) : (2 * i + 1); swap(vec[i], vec[temp]); i = temp; } else if (vec[i] < vec[2 * i + 1]) { swap(vec[i], vec[2 * i + 1]); i = 2 * i + 1; } else if (vec[i] < vec[2 * i + 2]) { swap(vec[i], vec[2 * i + 2]); i = 2 * i + 2; } else break; } else { if (vec[i] < vec[2 * i + 1]) { swap(vec[i], vec[2 * i + 1]); i = 2 * i + 1; } else break; } } else break; } }
|