xmengnet/hello-algo
1
0
Fork 0
mirror of https://github.com/krahets/hello-algo.git synced 2024-12-26 02:36:28 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

update zig codes for Section 'Heap' (my_heap.zig)

Browse source
This commit is contained in:
sjinzh 2023-01-16 19:32:12 +08:00
parent 67f5050ca4
commit cb0a3f868d
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
codes/zig/chapter_heap/my_heap.zig
View file

@ -7,7 +7,7 @@ const inc = @import("include");
// 堆类简易实现 // 堆类简易实现
// 编译期泛型 // 编译期泛型
pub fn MyMaxHeap(comptime T: type) type { pub fn MaxHeap(comptime T: type) type {
return struct { return struct {
const Self = @This(); const Self = @This();
@ -17,7 +17,7 @@ pub fn MyMaxHeap(comptime T: type) type {
pub fn init(self: *Self, allocator: std.mem.Allocator, nums: []const T) !void { pub fn init(self: *Self, allocator: std.mem.Allocator, nums: []const T) !void {
if (self.maxHeap != null) return; if (self.maxHeap != null) return;
self.maxHeap = std.ArrayList(T).init(allocator); self.maxHeap = std.ArrayList(T).init(allocator);
// 所有元素入堆 // 将列表元素原封不动添加进堆
try self.maxHeap.?.appendSlice(nums); try self.maxHeap.?.appendSlice(nums);
// 堆化除叶结点以外的其他所有结点 // 堆化除叶结点以外的其他所有结点
var i: usize = parent(self.size() - 1) + 1; var i: usize = parent(self.size() - 1) + 1;
@ -155,7 +155,7 @@ pub fn main() !void {
const mem_allocator = mem_arena.allocator(); const mem_allocator = mem_arena.allocator();
// 初始化大顶堆 // 初始化大顶堆
var maxHeap = MyMaxHeap(i32){}; var maxHeap = MaxHeap(i32){};
try maxHeap.init(std.heap.page_allocator, &[_]i32{ 9, 8, 6, 6, 7, 5, 2, 1, 4, 3, 6, 2 }); try maxHeap.init(std.heap.page_allocator, &[_]i32{ 9, 8, 6, 6, 7, 5, 2, 1, 4, 3, 6, 2 });
defer maxHeap.deinit(); defer maxHeap.deinit();
std.debug.print("\n输入列表并建堆后\n", .{}); std.debug.print("\n输入列表并建堆后\n", .{});