go 算法 查詢字符在字符串中的位置

在utf8字符串判斷是否包含指定字符串，並返回下標。
"北京天安門最美麗" , "天安門"
結果：2

解答：

import (
    "fmt"
    "strings"
)

func main(){
    fmt.Println(Utf8Index("北京天安門最美麗", "天安門"))
    fmt.Println(strings.Index("北京天安門最美麗", "男"))
    fmt.Println(strings.Index("", "男"))
    fmt.Println(Utf8Index("12ws北京天安門最美麗", "天安門"))
}

func Utf8Index(str, substr string) int {
    asciiPos := strings.Index(str, substr)
    if asciiPos == -1 || asciiPos == 0 {
        return asciiPos
    }
    pos := 0
    totalSize := 0
    reader := strings.NewReader(str)
    for _, size, err := reader.ReadRune(); err == nil; _, size, err = reader.ReadRune() {
        totalSize += size
        pos++
        // 匹配到
        if totalSize == asciiPos {
            return pos
        }
    }
    return pos
}