每日一题---1816. 截断句子[力扣][Go]

func truncateSentence(s string, k int) string {
    // 空格出现的次数
    number := 0
    for i, sChar := range s {
        if sChar == ' ' {
            number ++
        }
        if number == k {
            number = i
            break
        }
    }
    // 如果空格出现次数小于单词数，返回原字符串
    if number < k {
        return s
    }
    return s[:number]
}