motiejus/zig
1
Fork 0
Code Releases Activity

std: Fix update() method in PriorityQueue and PriorityDequeue (#13908)

Browse Source 
Previously the update() method would iterate over its capacity, which may contain uninitialized memory or already removed elements.
This commit is contained in:
Frechdachs
2022-12-13 20:55:41 +01:00
committed by GitHub
parent 2e66b3be6e
commit 478544239e
2 changed files with 26 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/std/priority_queue.zig
View File

@@ -218,8 +218,10 @@ pub fn PriorityQueue(comptime T: type, comptime Context: type, comptime compareF
pub fn update(self: *Self, elem: T, new_elem: T) !void {
const update_index = blk: {
for (self.items) |item, idx| {
if (compareFn(self.context, item, elem).compare(.eq)) break :blk idx;
var idx: usize = 0;
while (idx < self.len) : (idx += 1) {
const item = self.items[idx];
if (compareFn(self.context, item, elem) == .eq) break :blk idx;
}
return error.ElementNotFound;
};
@@ -591,6 +593,15 @@ test "std.PriorityQueue: update same max heap" {
try expectEqual(@as(u32, 1), queue.remove());
}
test "std.PriorityQueue: update after remove" {
var queue = PQlt.init(testing.allocator, {});
defer queue.deinit();
try queue.add(1);
try expectEqual(@as(u32, 1), queue.remove());
try expectError(error.ElementNotFound, queue.update(1, 1));
}
test "std.PriorityQueue: siftUp in remove" {
var queue = PQlt.init(testing.allocator, {});
defer queue.deinit();