[LeetCode&Python] Problem 557. Reverse Words in a String III

Given a string, you need to reverse the order of characters in each word within a sentence while still preserving whitespace and initial word order.

Example 1:

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

Note: In the string, each word is separated by single space and there will not be any extra space in the string.

My first solution:

class Solution:
    def reverseWords(self, s):
        """
        :type s: str
        :rtype: str
        """
        answer=''
        smallstr=''

        for c in s:
            if c==' ':
                smallstr=smallstr[::-1]
                answer=answer+smallstr+' '
                smallstr=''
            else:
                smallstr=smallstr+c

        return answer+smallstr[::-1

　　

A much shorter solution:

class Solution:
    def reverseWords(self, s):
        """
        :type s: str
        :rtype: str
        """

        return ' '.join(i[::-1] for i in s.split(' '))