Submission #4467379

---

Source Code Expand

def d_precipitation(N, Precipitation_periods):
    import itertools

    # 降雨を5分刻みにする
    rainfall = []
    for period in Precipitation_periods:
        # period: 'hhmm-hhmm' (降り始め時刻-降り終わり時刻)
        # start/end: 開始/終了 時刻(0時0分から何分経ったか)
        start_hhmm, end_hhmm = map(int, period.split('-'))
        start = (start_hhmm // 100) * 60 + (start_hhmm % 100)
        end = (end_hhmm // 100) * 60 + (end_hhmm % 100)
        start -= start % 5  # 直前の5分単位に丸める
        end += (5 - (end % 5)) % 5  # 直後の5分単位に丸める
        rainfall.append((start, end))

    precipitation = [0] * (1440 + 1 + 1)  # 1日を1分単位で表す. 24時に+1, 番兵に+1
    for start, end in rainfall:
        precipitation[start] += 1
        precipitation[end + 1] -= 1
    precipitation = itertools.accumulate(precipitation)

    # 開始終了時刻を復元
    ans_tmp = []
    is_zero = True  # Falseなら1以上の値が現れた
    start, end = 0, 0
    for i, t in enumerate(precipitation):
        if t > 0 and is_zero:  # 0でない値が現れたので開始時刻とする
            start = i
            is_zero = False
        if t == 0 and not is_zero:  # 0が現れたのでその1つ前を終了時刻とする
            end = i - 1
            is_zero = True
            string = '{0:02d}{1:02d}-{2:02d}{3:02d}'
            str_format = (start // 60, start % 60, end // 60, end % 60)
            ans_tmp.append(string.format(*str_format))
    ans = '\n'.join(ans_tmp)
    return ans

N = int(input())
Precipitation_periods = [input() for _ in range(N)]
print(d_precipitation(N, Precipitation_periods))

Submission Info

Judge Result