Go channel 实现归并排序中的 merge 函数

最新推荐文章于 2025-04-22 03:11:39 发布

转载最新推荐文章于 2025-04-22 03:11:39 发布 · 291 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://juejin.im/post/5a6f3dbe6fb9a01c9332f26f

文章标签：

#python #ruby

本文介绍了一种使用Go语言的channel和Goroutine特性来实现归并排序的方法。通过定义Merge函数，该函数接收两个有序的channel，并返回一个新的有序channel。这种方式不仅展示了Go语言的并发特性，还提供了一个有趣的排序实现案例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近学习 Go，但是苦于没有项目练手，于是便逼迫自己：如果想到什么有趣的东西，看能不能用 Go 实现一遍，于是便有了这篇流水文。

实现过程

归并排序中的 merge 函数，相信每个人都很熟悉，网上随便搜搜都有一大堆文章，这里不再赘述细节。一开始，我用的是常规套路，不过觉得没啥意思，无非是「换汤不换药，感觉还是在拿自己熟悉的语言写东西」。

联想到 Go 的 channel 似乎能在某种程度上满足我的要求，再加上 Goroutine 这种东西，便想：是不是也能利用利用这两个语言特性。

channel 这个数据结构，在 Go 中有比较丰富的含义，但我基本上把它当队列使用。Goroutine 也一样，我基本把它等同于「用户态线程」（两者都很牛逼，不过作为应用层的开发者，有时候并不想深究太多，一切都往简单方向理解）。

由于我只是练手，所以我想到的 API 长这样：

Merge(ch1, ch2): outChan
复制代码

给定两个有序的 channel，然后将其合并为一个有序的 channel。

于是我的实现如下：


func Merge(ch1 <-chan int, ch2 <-chan int) <-chan int {

    out := make(chan int)

    go func() {
        // 等上游的数据 （这里有阻塞，和常规的阻塞队列并无不同）
        v1, ok1 := <-ch1
        v2, ok2 := <-ch2
        
        // 取数据
        for ok1 || ok2 {
            if !ok2 || (ok1 && v1 <= v2) {
                // 取到最小值, 就推到 out 中
                out <- v1
                v1, ok1 = <-ch1
            } else {
                out <- v2
                v2, ok2 = <-ch2
            }
        }
        // 显式关闭
        close(out)
    }()

    // 开完goroutine后, 主线程继续执行, 不会阻塞
    return out
}
复制代码