557. 反转字符串中的单词 III

给定一个字符串,你需要反转字符串中每个单词的字符顺序,同时仍保留空格和单词的初始顺序。

示例 1:

输入: "Let's take LeetCode contest"
输出: "s'teL ekat edoCteeL tsetnoc" 

注意:在字符串中,每个单词由单个空格分隔,并且字符串中不会有任何额外的空格。

思路:

用一个loc记录空格的下一个位置,loc初始化为0,没遇到一个空格,就将loc到空格之间的字符串反转。

代码:

class Solution {
public:
    string reverseWords(string s) {
        int loc=0;
        char temp;
        for(int i=0;i<s.length();i++)
        {
            if(s[i]==' ')                  //没遇到空格,就将loc和空格之间的字符串反转
            {
                for(int k=loc,p=1;k<(loc+i)/2;k++)
                {
                    temp=s[k];
                    s[k]=s[i-p];
                    s[i-p]=temp;
                    p++;
                }
                loc=i+1;                   //反转完后loc更新为空格的下一个位置
            }
            if(i==s.length()-1)            //因为结尾不是空格,所以结尾需要另外判断
            {
                for(int k=loc,p=0;k<(loc+i+1)/2;k++)
                {
                    temp=s[k];
                    s[k]=s[i-p];
                    s[i-p]=temp;
                    p++;
                }
            }
        }
        return s;
    }
};

效率: