diff --git a/codes/zig/build.zig b/codes/zig/build.zig index 4299c60f2..85119b8fd 100644 --- a/codes/zig/build.zig +++ b/codes/zig/build.zig @@ -187,7 +187,21 @@ pub fn build(b: *std.build.Builder) void { if (b.args) |args| run_cmd_binary_tree.addArgs(args); const run_step_binary_tree= b.step("run_binary_tree", "Run binary_tree"); run_step_binary_tree.dependOn(&run_cmd_binary_tree.step); - + + // Section: "Heap" + // Source File: "chapter_heap/heap.zig" + // Run Command: zig build run_heap + const exe_heap = b.addExecutable("heap", "chapter_heap/heap.zig"); + exe_heap.addPackagePath("include", "include/include.zig"); + exe_heap.setTarget(target); + exe_heap.setBuildMode(mode); + exe_heap.install(); + const run_cmd_heap = exe_heap.run(); + run_cmd_heap.step.dependOn(b.getInstallStep()); + if (b.args) |args| run_cmd_heap.addArgs(args); + const run_step_heap = b.step("run_heap", "Run heap"); + run_step_heap.dependOn(&run_cmd_heap.step); + // Section: "Linear Search" // Source File: "chapter_searching/linear_search.zig" // Run Command: zig build run_linear_search diff --git a/codes/zig/chapter_heap/heap.zig b/codes/zig/chapter_heap/heap.zig new file mode 100644 index 000000000..1f4162a53 --- /dev/null +++ b/codes/zig/chapter_heap/heap.zig @@ -0,0 +1,84 @@ +// File: heap.zig +// Created Time: 2023-01-14 +// Author: sjinzh (sjinzh@gmail.com) + +const std = @import("std"); +const inc = @import("include"); + +fn lessThan(context: void, a: i32, b: i32) std.math.Order { + _ = context; + return std.math.order(a, b); +} + +fn greaterThan(context: void, a: i32, b: i32) std.math.Order { + return lessThan(context, a, b).invert(); +} + +fn testPush(comptime T: type, mem_allocator: std.mem.Allocator, heap_push: anytype, val: T) !void { + var heap = heap_push; + try heap.add(val); //元素入堆 + std.debug.print("\n元素 {} 入堆后\n", .{val}); + try inc.PrintUtil.printHeap(T, mem_allocator, heap); +} + +fn testPop(comptime T: type, mem_allocator: std.mem.Allocator, heap_pop: anytype) !void { + var val = heap_pop.remove(); //堆顶元素出堆 + std.debug.print("\n堆顶元素 {} 出堆后\n", .{val}); + try inc.PrintUtil.printHeap(T, mem_allocator, heap_pop); +} + +// Driver Code +pub fn main() !void { + // 初始化内存分配器 + var mem_arena = std.heap.ArenaAllocator.init(std.heap.page_allocator); + defer mem_arena.deinit(); + const mem_allocator = mem_arena.allocator(); + + // 初始化堆 + // 初始化小顶堆 + const PQlt = std.PriorityQueue(i32, void, lessThan); + var minHeap = PQlt.init(std.heap.page_allocator, {}); + defer minHeap.deinit(); + // 初始化大顶堆 + const PQgt = std.PriorityQueue(i32, void, greaterThan); + var maxHeap = PQgt.init(std.heap.page_allocator, {}); + defer maxHeap.deinit(); + + std.debug.print("\n以下测试样例为大顶堆", .{}); + + // 元素入堆 + try testPush(i32, mem_allocator, &maxHeap, 1); + try testPush(i32, mem_allocator, &maxHeap, 3); + try testPush(i32, mem_allocator, &maxHeap, 2); + try testPush(i32, mem_allocator, &maxHeap, 5); + try testPush(i32, mem_allocator, &maxHeap, 4); + + // 获取堆顶元素 + var peek = maxHeap.peek().?; + std.debug.print("\n堆顶元素为 {}\n", .{peek}); + + // 堆顶元素出堆 + try testPop(i32, mem_allocator, &maxHeap); + try testPop(i32, mem_allocator, &maxHeap); + try testPop(i32, mem_allocator, &maxHeap); + try testPop(i32, mem_allocator, &maxHeap); + try testPop(i32, mem_allocator, &maxHeap); + + // 获取堆的大小 + var size = maxHeap.len; + std.debug.print("\n堆元素数量为 {}\n", .{size}); + + // 判断堆是否为空 + var isEmpty = if (maxHeap.len == 0) true else false; + std.debug.print("\n堆是否为空 {}\n", .{isEmpty}); + + // 输入列表并建堆 + // 时间复杂度为 O(n) ,而非 O(nlogn) + try minHeap.addSlice(&[_]i32{ 1, 3, 2, 5, 4 }); + std.debug.print("\n输入列表并建立小顶堆后\n", .{}); + try inc.PrintUtil.printHeap(i32, mem_allocator, minHeap); + + const getchar = try std.io.getStdIn().reader().readByte(); + _ = getchar; +} + diff --git a/codes/zig/include/TreeNode.zig b/codes/zig/include/TreeNode.zig index 7c1e2ad22..a3715dbdf 100644 --- a/codes/zig/include/TreeNode.zig +++ b/codes/zig/include/TreeNode.zig @@ -28,32 +28,26 @@ pub fn arrToTree(comptime T: type, mem_allocator: std.mem.Allocator, list: []T) if (list.len == 0) return null; var root = try mem_allocator.create(TreeNode(T)); root.init(list[0]); - - const TailQueue = std.TailQueue(?*TreeNode(T)); - const TailQueueNode = std.TailQueue(?*TreeNode(T)).Node; - var que = TailQueue{}; - var node_root = TailQueueNode{ .data = root }; - que.append(&node_root); + var que = std.ArrayList(*TreeNode(T)).init(std.heap.page_allocator); + try que.append(root); var index: usize = 0; - while (que.len > 0) { - var node = que.popFirst(); + while (que.items.len > 0) { + var node = que.orderedRemove(0); index += 1; if (index >= list.len) break; if (index < list.len) { var tmp = try mem_allocator.create(TreeNode(T)); tmp.init(list[index]); - node.?.data.?.left = tmp; - var a = TailQueueNode{ .data = node.?.data.?.left }; - que.append(&a); + node.left = tmp; + try que.append(node.left.?); } index += 1; if (index >= list.len) break; if (index < list.len) { var tmp = try mem_allocator.create(TreeNode(T)); tmp.init(list[index]); - node.?.data.?.right = tmp; - var a = TailQueueNode{ .data = node.?.data.?.right }; - que.append(&a); + node.right = tmp; + try que.append(node.right.?); } } return root;