LeetCode之Reverse Words in a String III

题目概述：

给定一个字符串，把其中的每个单词翻转过来，保留原有空格和单词顺序。

思路：

用STL中的reverse()函数。

示例：

Input: "Let's take LeetCode contest"
Output: "s'teL ekat edoCteeL tsetnoc"

代码：

class Solution {
public:
    string reverseWords(string s) {
        string str1,str2;//str1用于存放s中的单词；str2用于存放最终的倒序字符串
        for(int i=0,j=0;i<s.length();i++,j++)
        {
           //空格前的单词处理
           if(s[i]!=' ')
           {
               str1.push_back(s[i]);
           }
           else
           {
               reverse(str1.begin(),str1.end());//单词逆序
               str2+=str1+' ';
               str1.erase();
           }
           //对于最后的单词处理
           if(i==s.length()-1)
           {
               reverse(str1.begin(),str1.end());
               str2+=str1;
               str1.erase();
           }
        }
        return str2;
    }
};