1. 程式人生 > >Leetcode 345. 反轉字串中的母音字母 By Python

Leetcode 345. 反轉字串中的母音字母 By Python

編寫一個函式,以字串作為輸入,反轉該字串中的母音字母。

示例 1:

輸入: "hello"
輸出: "holle"

示例 2:

輸入: "leetcode"
輸出: "leotcede"

說明:
母音字母不包含字母"y"。


思路

設立2個指標,一個從索引0開始向右,一個從末尾向前,根據條件進行處理即可

程式碼

class Solution:
    def reverseVowels(self, s):
        """
        :type s: str
        :rtype: str
        """
        yuan = ['a','e','i','o','u','A','E','I','O','U']
        s = list(s)
        
        l = 0
        r = len(s)-1
        while l < r:
            if s[l] in yuan and s[r] in yuan:
                s[l], s[r] = s[r], s[l]
                l += 1
                r -= 1
            elif s[l] in yuan and s[r] not in yuan:
                r -= 1
            elif s[l] not in yuan and s[r] in yuan:
                l += 1
            else:
                l += 1
                r -= 1
        return ''.join(s)