题目描述

给定一个字符串 text，该字符串由若干被空格包围的单词组成。每个单词由一个或者多个小写英文字母组成，并且两个单词之间至少存在一个空格。题目测试用例保证 text 至少包含一个单词。

请你重新排列空格，使每对相邻单词之间的空格数目都 相等，并尽可能 最大化 该数目。如果不能重新平均分配所有空格，请 将多余的空格放置在字符串末尾 ，这也意味着返回的字符串应当与原 text 字符串的长度相等。

返回 重新排列空格后的字符串。

样例

输入：text = "  this   is  a sentence "
输出："this   is   a   sentence"
解释：总共有 9 个空格和 4 个单词。
可以将 9 个空格平均分配到相邻单词之间，相邻单词间空格数为：9 / (4-1) = 3 个。

输入：text = " practice   makes   perfect"
输出："practice   makes   perfect "
解释：总共有 7 个空格和 3 个单词。
7 / (3-1) = 3 个空格加上 1 个多余的空格。多余的空格需要放在字符串的末尾。

输入：text = "hello   world"
输出："hello   world"

输入：text = "  walks  udp package   into  bar a"
输出："walks  udp  package  into  bar  a "

输入：text = "a"
输出："a"

限制

• 1 <= text.length <= 100
• text 由小写英文字母和 ' ' 组成
• text 中至少包含一个单词

算法

(模拟) $O(n)$

1. 按照题目描述模拟，注意只有一个单词的情况。

时间复杂度

• 遍历字符串若干次，故总时间复杂度为 $O(n)$。

空间复杂度

• 需要 $O(n)$ 的额外空间存储答案。

Go 代码

func reorderSpaces(text string) string {
    words := strings.Split(text, " ")
    realWords := make([]string, 0)

    for _, word := range words {
        if len(word) > 0 {
            realWords = append(realWords, word)
        }
    }
    numOfWords := len(realWords)

    numOfWhitespace := 0
    for _, v := range text {
        if v == ' ' {
            numOfWhitespace++
        }
    }

    var res bytes.Buffer
    if numOfWords == 1 {
        res.WriteString(realWords[0])
        for i := 0; i < numOfWhitespace; i++ {
            res.WriteString(" ")
        }
        return res.String()
    }

    gap := numOfWhitespace / (numOfWords - 1)
    for i, word := range realWords {
        res.WriteString(word)
        if i == numOfWords - 1 {
            break
        }
        for j := 0; j < gap; j++ {
            res.WriteString(" ")
        }
    }

    for i := 0; i < numOfWhitespace % (numOfWords - 1); i++ {
        res.WriteString(" ")
    }
    return res.String()
}