【Leetcode】1592. 重新排列单词间的空格

1592. 重新排列单词间的空格

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

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

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

示例 1：

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

示例 2：

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

示例 3：

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

示例 4：

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

示例 5：

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

提示：

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

---

解题思路

这题的思路还是比较简单的，重新排列空格，使每个单词之间的空格数目相同且最大，只需要求出空格数目和取出所有的单词，然后对但单词间的间隔数取整就行，多余的空格加到字符串尾部就行。可能难点在于不知道如何取出所有的单词，需要你了解stringstream的用法。如果不了解stringstream，可以看看这篇文章https://zhuanlan.zhihu.com/p/44435521

本题大致可以分为三步来做。

1. 求出所有的空格数目。遍历字符串就可以求出。
2. 求出所有的单词。用stingstream，取出所有的单词，存到vector中。
3. 分析单词的数目
   • 只有一个单词，单独处理，将所有的空格全部加入到尾部就可以。
   • 多个 单词，计算出单词间隔的空格数目，然后依次连接，多余的空格全部加到尾部。

---

代码

class Solution
{
public:
    string reorderSpaces(string text)
    {
        size_t n = text.size();
        int cnt_space = 0;
        //统计空格数
        for (size_t i = 0; i < n; i++)
        {
            if (text[i] == ' ')
                cnt_space++;
        }
        //取出所有单词并且保存
        stringstream in(text);
        string word;
        vector<string> words;
        while (in >> word)
        {
            words.push_back(word);
        }

        string res;
        //只存在一个单词
        if (words.size() == 1)
        {
            //尾部空格最后处理
            res = words[0];
            //加入尾部空格
            string tailspace(cnt_space, ' ');
            return res + tailspace;
        }
        //存在多个单词
        else
        {

            //每个单词间的空格数量
            int x = cnt_space / (words.size() - 1);
            //初始化空格字符串
            string space(x, ' ');
            //形成答案字符串，最后一个单词和尾部空格最后处理
            for (int i = 0; i < words.size() - 1; i++)
            {
                res += words[i] + space;
            }
             //加入尾部空格
            string tailspace(cnt_space % (words.size() - 1), ' ');
            return res + words.back() + tailspace;
        }
    }
};