結果

問題 No.2408 Lakes and Fish
ユーザー ryomacryomac
提出日時 2023-08-11 21:45:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 341 ms / 2,000 ms
コード長 672 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,664 KB
実行使用メモリ 91,768 KB
最終ジャッジ日時 2024-11-18 15:46:37
合計ジャッジ時間 5,995 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #

from bisect import bisect_left, bisect_right

N, M = map(int, input().split())
L = list(map(int, input().split()))
FBW = [list(map(int, input().split())) for _ in range(M)]

ans = 0
for i in range(M):
    f, b, w = FBW[i]
    idx = bisect_left(L, f)
    if idx != N and L[idx] == f:
        ans += w
    # if f in L:
    # ans += w
    else:
        idx = bisect_left(L, f)
        if idx == 0:
            cost = L[0] - f
        elif idx == N:
            cost = f - L[-1]
        else:
            near = L[bisect_left(L, f)]
            near2 = L[bisect_right(L, f) - 1]
            cost = min(abs(f - near), abs(f - near2))
        ans += max(w - cost, b)
print(ans)
0