Go语言面试题:交替打印数字和字母

3,312 阅读1分钟

问题描述

使⽤两个goroutine交替打印序列,⼀个goroutine打印数字, 另外⼀个goroutine打印字⺟, 最终效果如下:

12AB34CD56EF78GH910IJ1112KL1314MN1516OP1718QR1920ST2122UV2324WX2526YZ2728

解题思路

就像我们在操作系统中学到的用信号量来控制进程协作一样,我们可以用两个无缓冲channel来控制goroutine的协作。

一个叫做numberchannel用来通知打印数字的goroutine

另一个叫做letterchannel用来通知打印字母的goroutine

number, letter := make(chan bool), make(chan bool)

打印数字的goroutine内容很简单:先等待number的管道通知,然后打印数字,最后通知letter管道。

go func() {
            i := 1
            for {
                    <- number
                    fmt.Printf("%d%d", i, i + 1)
                    i += 2
                    letter <- true
            }
}()

打印字母的goroutine的内容也是类似,不过需要控制结束条件:

go func() {
            str := "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
            i := 0
            for {
                    <- letter
                    if i >= utf8.RuneCountInString(str) {
                            return
                    }
                    fmt.Print(str[i : i+2])
                    i += 2
                    number <- true
            }
}()

完整代码

最后,我们加上syn.WaitGroup来让主协程等待打印协程的工作结束,完整代码如下:

package main

import (
        "fmt"
        "sync"
        "unicode/utf8"
)

func main() {
        number, letter := make(chan bool), make(chan bool)
        wait := sync.WaitGroup{}
        go func() {
                i := 1
                for {
			<- number
                        fmt.Printf("%d%d", i, i + 1)
                        i += 2
                        letter <- true
                }
        }()
        wait.Add(1)
        go func(wait *sync.WaitGroup) {
                str := "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
                i := 0
                for {
                        <- letter
                        if i >= utf8.RuneCountInString(str) {
                                wait.Done()
                                return
                        }
                        fmt.Print(str[i : i+2])
                        i += 2
                        number <- true
                }
        }(&wait)
        number <- true
        wait.Wait()
}