通过协程查找数组中元素

给定一个无序数组，求某个数值是否存在数组当中。

正常需要先排序，再二分去做的，不过由于 golang 中是可以很方便就可以起多个协程，所以测试下，多个协程分别查找各自的一段，如果查到就通知其他协程停止查找，当然为了完整性，在程序执行 5 秒，还未找到则停止查找。

package main

import (
    "context"
    "fmt"
    "math"
    "math/rand"
    "time"
)

func generate(size int) []int {

    return rand.Perm(size)
}

func generateTarget(size int) int {

    rand.Seed(time.Now().Unix())

    return int(rand.Int31n(int32(size)))
}

func main() {

    size := math.MaxInt32 / 1000

    arr := generate(size)
    target := generateTarget(size)

    n := len(arr)

    routineCount := 10 // 开 routineCount 个 goroutine

    sp := n / routineCount // 将数组切割成每份 sp 个

    ctx, cancel := context.WithCancel(context.Background()) // 利用 context 的特性

    ch := make(chan int) // 找到结果的通知 chan

    fmt.Println("开始查找")
    for i := 0; i < routineCount; i++ {
        go func(i int, ctx context.Context) {
            curArr := arr[i*sp : (i+1)*sp] // 每次切割一部分

            for _, v := range curArr {

                select {
                case <-ctx.Done(): // 如果通知结束，则终止
                    return
                default: // 其他情况直接去比较
                }
                if v == target {
                    ch <- target // 找到了将消息发送出来
                }
            }
        }(i, ctx)
    }

    fmt.Println("target", target) // 需要找的 targe
    select {
    case t := <-ch:
        fmt.Println("找到了", t)
        cancel() // 找到了，结束所有子 goroutine
    case <-time.After(5 * time.Second): // 有个倒计时
        fmt.Println("未找到")
    }
}

测试了下速度，确实很快

有疑问加站长微信联系（非本文作者）