重新排列单词间的空格

题目描述：
给你一个字符串 text ，该字符串由若干被空格包围的单词组成。每个单词由一个或者多个小写英文字母组成，并且两个单词之间至少存在一个空格。题目测试用例保证 text 至少包含一个单词 。
请你重新排列空格，使每对相邻单词之间的空格数目都 相等 ，并尽可能 最大化 该数目。如果不能重新平均分配所有空格，请 将多余的空格放置在字符串末尾 ，这也意味着返回的字符串应当与原 text 字符串的长度相等。
返回 重新排列空格后的字符串 。
示例 ：
输入：text = " this is a sentence "
输出：“this is a sentence”
解释：总共有 9 个空格和 4 个单词。可以将 9 个空格平均分配到相邻单词之间，相邻单词间空格数为：9 / (4-1) = 3 个。
提示：

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

解法

遍历即可，注意只有一个单词的情况，需要单独处理。

代码

class Solution:
    def reorderSpaces(self, text: str) -> str:
        word_list = []
        space_num = 0
        idx = 0
        while idx < len(text):
            while idx < len(text) and text[idx] == " ":
                idx += 1
                space_num += 1
            start = idx
            while idx < len(text) and text[idx] != " ":
                idx += 1
            if idx > start:
                word_list.append(text[start: idx])
        if len(word_list) == 1:
            return word_list[0] + " " * space_num
        m, n = space_num // (len(word_list) - 1), space_num % (len(word_list) - 1)
        return (" " * m).join(word_list) + " " * n
      

测试结果

执行用时：32 ms, 在所有 Python3 提交中击败了 55.66% 的用户
内存消耗：15 MB, 在所有 Python3 提交中击败了 41.87% 的用户

说明

算法题来源：力扣(LeetCode)