【Python】文字列の母音を先にソートし、その後に子音をソートして連結する方法

小文字のアルファベットのみで構成された文字列 s が与えられたとき、すべての母音をアルファベット順に並べたものの後ろにすべての子音をアルファベット順に並べたものを連結した文字列を作る問題を考えてみましょう。

問題例

入力が "helloworld" の場合、出力は "eoodhlllrw" になります。
これは、母音が e、o、o の3つでソート後も "eoo"、子音をソートすると "dhlllrw" になるためです。

解き方のアプローチ

この問題は、以下の手順で解くことができます。

1. 母音を格納するための空文字列 k と、子音を格納するための空文字列 t を用意します。
2. 文字列 s の各文字 c に対して次の処理を行います。
   • c が母音であれば、k に連結します。
   • それ以外の場合は、t に連結します。
3. ループが終わったら、k と t をそれぞれソートし、k + t の順に連結して返します。

実装例(Python)

理解を深めるために、実際のコードを見てみましょう。

class Solution:
    def solve(self, s):
        vowels = 'aeiou'
        k = ''
        t = ''
        for c in s:
            if c in vowels:
                k = k + c
            else:
                t = t + c
        k = ''.join(sorted(k))
        t = ''.join(sorted(t))
        return k + t

ob = Solution()
print(ob.solve("helloworld"))

入力

"helloworld"

出力

eoodhlllrw

計算量の目安

この解法では、まず文字列を一度走査して母音と子音を振り分け、その後それぞれをソートしています。文字列の長さを n とすると、時間計算量は O(n log n)、必要な追加メモリは O(n) となります。シンプルで直感的なアプローチであり、小規模な文字列であれば十分に高速に動作します。