Pythonでリストから目標値以上となる最初の値を見つける方法

数値のリスト rooms と目標値 t が与えられたとします。このとき、rooms の中で t 以上となる最初の値を見つける必要があります。該当する値が存在しない場合は -1 を返します。

例えば、入力が rooms = [20, 15, 35, 55, 30]、t = 30 の場合、出力は 35 になります。これは、リストを先頭から順に確認したときに、初めて 30 以上となる値が 35 だからです。それ以前の要素(20 や 15)は目標値 30 を満たしていません。

解決のアプローチ

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

• rooms 内の各要素 room を先頭から順に調べる
• room >= t を満たしていれば、その room を返す
• ループが終了しても見つからなければ、-1 を返す

実装例

理解を深めるために、次の実装例を見てみましょう。

def solve(rooms, t):
   for room in rooms:
      if room >= t:
         return room
   return -1

rooms = [20, 15, 35, 55, 30]
t = 30
print(solve(rooms, t))

入力

[20, 15, 35, 55, 30], 30

出力

35

計算量について

このアルゴリズムはリストを一度だけ走査する線形探索であるため、時間計算量は O(n)、空間計算量は O(1) となります(n はリストの要素数)。リストがソートされていなくても正しく動作する点が特徴です。

別の書き方:next() を使った簡潔な実装

ジェネレータ式と組み込み関数 next() を組み合わせると、同じ処理をより簡潔に記述できます。

def solve(rooms, t):
    return next((room for room in rooms if room >= t), -1)

第2引数の -1 は、条件を満たす要素がひとつも存在しなかった場合に返されるデフォルト値です。状況に応じて、可読性の高い for ループ版と、簡潔な next() 版を使い分けるとよいでしょう。