結果

問題 No.1950 片道きゃっちぼーる
ユーザー terasa
提出日時 2022-05-22 11:00:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,116 ms / 3,000 ms
コード長 944 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,272 KB
実行使用メモリ 377,592 KB
最終ジャッジ日時 2024-09-20 12:27:38
合計ジャッジ時間 13,518 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import pypyjit
import itertools
import heapq
import math
from collections import deque, defaultdict, Counter

input = sys.stdin.readline
sys.setrecursionlimit(10 ** 6)
pypyjit.set_param('max_unroll_recursion=-1')


N = int(input())
X = list(map(int, input().split()))
A = list(map(int, input().split()))

D = {}
for i, x in enumerate(X):
    D[x] = i

E = [[] for _ in range(N)]
for i, (x, a) in enumerate(zip(X, A)):
    l, r = x - a, x + a
    li, ri = D.get(l), D.get(r)
    if li is not None:
        E[li].append(i)
    if ri is not None:
        E[ri].append(i)

B = [(i, x + a) for i, x, a in zip(range(N), X, A)]
B.sort(key=lambda x: x[1], reverse=True)

ans = [None] * N
flag = [False] * N
for i, m in B:
    if flag[i] is True:
        continue

    def dfs(v):
        flag[v] = True
        ans[v] = m - X[v]
        for d in E[v]:
            if flag[d] is False:
                dfs(d)
    dfs(i)
print(*ans, sep='\n')
0