LeetCode 筆記 - 2785. Sort Vowels in a String

class Solution:
    def sortVowels(self, s: str) -> str:

        if (size := len(s)) == 1:
            return s

        vowels = set('aeiouAEIOU')
        s_vowels = [c for c in s if c in vowels]
        s_vowels.sort()

        vowel_index = 0
        result = ''
        for c in s:
            if c in vowels:
                result += s_vowels[vowel_index]
                vowel_index += 1
            else:
                result += c

        return result