Golang面试题: 3个goroutine交替打印ABC

文章目录

开心一刻

       一位小帅哥和靓姑娘在火车上相遇。经过一阵“天南地北,生猛海鲜”之后,小帅哥拿出一副扑克与姑娘对玩:
  帅哥:QQK?(谈谈看)
  姑娘:Q45?(谈什么)
  帅哥:Q21!(谈恋爱)
  姑娘:8Q!!(不谈)
  $%……!
  快到站了,小帅哥不甘心,又拿出扑克:
  帅哥:3QQK?(再谈谈看)
  姑娘:948Q!!!(就是不谈)
  帅哥:—%¥……)

题目介绍

       使用3个goroutine交替打印ABC,并且在输出是也按照ABCABC输出

做法一

package main

import (
	"fmt"
	"sync"
)
func main() {
	var ch1, ch2, ch3 = make(chan struct{}), make(chan struct{}), make(chan struct{})
	var wg sync.WaitGroup
	wg.Add(3)
	go func(s string) {
		defer wg.Done()
		for i := 1; i <= 10; i++ {
			<- ch1
			fmt.Print(s)
			ch2 <- struct{}{}
		}
		<- ch1
	}("A")
	go func(s string) {
		defer wg.Done()
		for i := 1; i <= 10; i++ {
			<- ch2
			fmt.Print(s)
			ch3 <- struct{}{}
		}
	}("B")
	go func(s string) {
		defer wg.Done()
		for i := 1; i <= 10; i++ {
			<- ch3
			fmt.Println(s)
			ch1 <- struct{}{}
		}
	}("C")
	ch1 <- struct{}{}
	wg.Wait()
}

       由同步channel控制线程之间的执行顺序即可。

题目扩展

使用N个协程交替打印英文字母

package main

import "fmt"

func main() {
	chanNum := 4
	chanQueue := make([]chan struct{}, chanNum)
	var result = 0
	exitChan := make(chan struct{})
	for i := 0; i < chanNum; i++ {
		chanQueue[i] = make(chan struct{})
		if i == chanNum - 1 {
			go func(i int) {
				chanQueue[i] <- struct{}{}
			}(i)
		}
	}
	for i := 0; i < chanNum; i++ {
		var lastChan, curChan chan struct{}
		if i == 0 {
			lastChan = chanQueue[chanNum - 1]
		} else {
			lastChan = chanQueue[i - 1]
		}
		curChan = chanQueue[i]
		go func(i byte, lastChan, curChan chan struct{}) {
			for {
				if result > 20 {
					exitChan <- struct{}{}
				}
				<- lastChan
				fmt.Printf("%c\n", i)
				result++
				curChan <- struct{}{}
			}
		}('A'+byte(i), lastChan, curChan)
	}
	<- exitChan
	fmt.Println("done")
}
  1. 第一个for循环中为chanQueue数组中的每个channel实例化。同时给最后一个channel写一条数组,为了第一次输出能从第一个channel中输出,如果不写会造成死锁,而且需要创建一个协程来写,否则程序直接报错。
  2. 第二个for循环中lastChan意思是,如果当前chan要打印数据,必须是上一个chan打印完后才能打印。
  3. 在这里使用匿名函数的时候最好使用这种带参数的传参,而不是直接使用全局变量,否则因为协程并发执行的不确定性可能造成死锁。

文中都是我个人的理解,如有错误的地方欢迎下方评论告诉我,我及时更正,大家共同进步

上一篇:Golang-协程、管道和文件操作综合练习题


下一篇:go chan练习:等价二叉查找树