Pythonで文を切り詰めて最初のk個の単語を抽出するプログラムの書き方

問題の概要

単一のスペースで区切られた英語の単語を含む文 s があるとします(先頭や末尾に余分なスペースはないものとします)。さらに、整数値 k も与えられます。このとき、文を切り詰めて、最初の k 個の単語のみを取得する必要があります。

例

例えば、入力が以下のような場合を考えてみましょう。

• s = "Coding challenges are really helpful for students"
• k = 5

この場合、出力は次のようになります。

Coding challenges are really helpful

解決アプローチ

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

1. split() メソッドを使って、文 s をスペースで分割し、単語のリストを作成します。
2. スライス構文 words[:k] で、リストの最初の k 個の単語を取り出します。
3. join() メソッドで、取り出した単語をスペース区切りの文字列として連結して返します。

実装例

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

def solve(s, k):
    words = s.split()
    return " ".join(words[:k])

s = "Coding challenges are really helpful for students"
k = 5
print(solve(s, k))

入力

"Coding challenges are really helpful for students", 5

出力

Coding challenges are really helpful

コードの解説

s.split() は引数なしで呼び出すと、連続する空白文字を基準に文字列を分割し、単語のリストを返します。そのため、仮に先頭や末尾に空白が含まれていても、空の要素が生じることなく正しく処理できます。

words[:k] はリストのスライス操作で、インデックス0からk-1までの要素、つまり最初のk個の単語を取得します。最後に " ".join(...) を使うことで、各単語を半角スペースで連結した文字列を生成して返しています。

このアルゴリズムの計算量は、文の長さを n とすると、時間計算量・空間計算量ともに O(n) となり、非常に効率的です。文章の要約表示やプレビュー機能など、テキストの先頭部分だけを取り出したい場面で幅広く活用できるテクニックです。