Pythonで特定の時刻に作業中だったプログラマーの人数を求める方法

区間（インターバル）のリストと、ある特定の時刻 time が与えられているとします。各区間は [start, end] という構造を持ち、それぞれのプログラマーが作業していた時間帯を表しています。このとき、指定した時刻に実際に作業していたプログラマーの人数を求めるのがこの問題です。

例として、入力が interval = [[2, 6], [4, 10], [5, 9], [11, 14]]、time = 5 である場合を考えてみましょう。時刻 5 において作業中の区間は [2, 6]、[4, 10]、[5, 9] の3つであるため、出力は 3 となります。

解法のアプローチ

この問題は非常にシンプルで、以下の手順で解くことができます。

• カウンター変数 count を 0 で初期化する。
• 区間リスト内の各インターバルに対して以下を繰り返す。
  • 区間の開始時刻 start が time 以下であり、かつ終了時刻 end が time 以上であれば、count を 1 増やす。
• 最後に count を返す。

この方法では各区間を1回ずつチェックするだけなので、計算量は O(n) となり、効率的です。

実装例

それでは、実際のコードを見て理解を深めましょう。

サンプルコード

class Solution:
    def solve(self, intervals, time):
        count = 0
        for interval in intervals:
            if interval[0] <= time and interval[1] >= time:
                count += 1
        return count

ob = Solution()
interval = [[2, 6], [4, 10], [5, 9], [11, 14]]
time = 5
print(ob.solve(interval, time))

入力

[[2, 6], [4, 10], [5, 9], [11, 14]], 5

出力

3

まとめ

このように、与えられた時刻が各区間の範囲内（start ≤ time ≤ end）に含まれているかどうかを順番に判定していくだけで、簡単に該当するプログラマーの人数を数えることができます。条件判定には不等号を使い、境界値（ちょうど開始時刻や終了時刻と一致する場合）もカウント対象に含めている点に注意しましょう。