結果

問題 No.1965 Heavier
ユーザー atamic
提出日時 2025-02-14 12:03:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 461 ms / 2,000 ms
コード長 1,001 bytes
コンパイル時間 345 ms
コンパイル使用メモリ 82,036 KB
実行使用メモリ 150,152 KB
最終ジャッジ日時 2025-02-14 12:03:51
合計ジャッジ時間 10,978 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

import heapq

def solve():
    n = int(input())
    a_values = list(map(int, input().split()))
    b_values = list(map(int, input().split()))

    a = []
    for i in range(n):
        a.append((a_values[i], b_values[i]))

    a.sort()

    q1 = []
    q2 = []

    heapq.heappush(q1, (- (a[0][0] + a[0][1]), 0))
    heapq.heappush(q2, (- (a[0][0] - a[0][1]), 0))

    ans = 0
    l = 0
    for r in range(1, n):
        while q1:
            c_fi, c_se = -q1[0][0], q1[0][1]
            if a[r][0] + a[r][1] <= c_fi:
                heapq.heappop(q1)
                l = max(l, c_se + 1)
            else:
                break
        heapq.heappush(q1, (- (a[r][0] + a[r][1]), r))

        while q2:
            c_fi, c_se = -q2[0][0], q2[0][1]
            if a[r][0] - a[r][1] <= c_fi:
                heapq.heappop(q2)
                l = max(l, c_se + 1)
            else:
                break
        heapq.heappush(q2, (- (a[r][0] - a[r][1]), r))
        ans += r - l

    print(ans)

solve()
0