力扣面试150题 ｜151.反转字符串中的单词

题目描述

151.反转字符串中的单词

给你一个字符串 s ，请你反转字符串中 单词 的顺序。

单词 是由非空格字符组成的字符串。s 中使用至少一个空格将字符串中的 单词 分隔开。

返回 单词 顺序颠倒且 单词 之间用单个空格连接的结果字符串。

注意：输入字符串 s中可能会存在前导空格、尾随空格或者单词间的多个空格。返回的结果字符串中，单词间应当仅用单个空格分隔，且不包含任何额外的空格。

示例 1：

输入：s = “the sky is blue”
输出：“blue is sky the”

示例 2：

输入：s = " hello world "
输出：“world hello”
解释：反转后的字符串中不能存在前导空格和尾随空格。

示例 3：

输入：s = “a good example”
输出：“example good a”
解释：如果两个单词间有多余的空格，反转后的字符串需要将单词间的空格减少到仅有一个。

提示：

• 1 <= s.length <= 104
• s 包含英文大小写字母、数字和空格 ’ ’
• s 中 至少存在一个 单词

进阶：如果字符串在你使用的编程语言中是一种可变数据类型，请尝试使用 O(1) 额外空间复杂度的 原地 解法。

解题思路

整体思路是先反转整个字符串，再翻转每个单词。

有以下三个步骤：

1. 移除多余的空格
2. 翻转整个字符串
3. 翻转每个单词

代码实现

class Solution {
public:
	// 移除多余的空格
	void removeExtraSpace(string& s) { // 去除所有空格并在相邻单词中间添加空格
		int slow = 0;
		for (int fast = 0; fast < s.size(); fast++) {
			if (s[fast] != ' ') { // 遇到非空格就处理，即删除所有空格
				if (slow != 0) s[slow++] = ' '; // 给单词之间添加空格。slow != 0说明不是第一个单词，需要在单词前添加空格。
				while (s[fast] != ' ' && fast < s.size()) { // 补上单词，遇到空格说明单词结束
					s[slow++] = s[fast++];
			}
		}
		s.resize(slow); // slow的大小就是去掉空格后字符串的大小
	}
	
	// 翻转函数
	void reverse (string& s, int begin, int end) {
		for (int i = begin; j = end; i < j; i++; j--) {
			swap(s[i], s[j]);
		}
	}
	
	string reverseWords(string s) {
		removeExtraSpace();
		reverse(s, 0, s.size() - 1);

		// 翻转每个单词
		int start = 0;
		for (int i = 0; i < s.size(); i++) {
			if (i == s.size() || s[i] == ' ') {
				reverse(s, start, i - 1);
				start = i + 1;
			}
		}
		return s;
	}
};