結果

問題 No.1965 Heavier
ユーザー chineristAC
提出日時 2022-06-03 23:02:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 508 ms / 2,000 ms
コード長 1,610 bytes
コンパイル時間 238 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 145,228 KB
最終ジャッジ日時 2024-09-21 03:12:24
合計ジャッジ時間 9,465 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys,random,bisect
from collections import deque,defaultdict
from heapq import heapify,heappop,heappush
from itertools import permutations
from math import log,gcd

input = lambda :sys.stdin.readline().rstrip()
mi = lambda :map(int,input().split())
li = lambda :list(mi())


def solve(N,A,B):

    r = -1
    M = []
    del_M = []
    m = []
    del_m = []
    res = 0
    for l in range(N):
        if r==l-1:
            heappush(M,-(A[l]+B[l]))
            heappush(m,-(A[l]-B[l]))
            r = l
        while del_M and M[0]==del_M[0]:
            heappop(M)
            heappop(del_M)
        while del_m and m[0]==del_m[0]:
            heappop(m)
            heappop(del_m)
        while r!=N-1:
            if -M[0] < A[r+1]+B[r+1] and -m[0] < A[r+1]-B[r+1]:
                heappush(M,-(A[r+1]+B[r+1]))
                heappush(m,-(A[r+1]-B[r+1]))
                r += 1
            else:
                break
        
        #print(l,r,M,m)
        res += r-l
        heappush(del_M,-(A[l]+B[l]))
        heappush(del_m,-(A[l]-B[l]))

    return res

def brute(N,A,B):
    res = 0
    for l in range(N):
        for r in range(l+1,N):
            if all(A[i]+B[i] < A[i+1]+B[i+1] for i in range(l,r)) and all(A[i]-B[i] < A[i+1]-B[i+1] for i in range(l,r)):
                res += 1
    return res

while False:
    N = 3
    A = [random.randint(1,20) for i in range(N)]
    B = [random.randint(1,20) for i in range(N)]
    A.sort()
    print(A)
    print(B)
    print(solve(N,A,B),brute(N,A,B))
    assert solve(N,A,B)==brute(N,A,B)


N = int(input())
A = li()
B = li()

print(solve(N,A,B))
0