結果

問題 No.1950 片道きゃっちぼーる
ユーザー terasaterasa
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
55,280 KB
testcase_01 AC 42 ms
55,208 KB
testcase_02 AC 40 ms
56,124 KB
testcase_03 AC 749 ms
371,060 KB
testcase_04 AC 806 ms
370,328 KB
testcase_05 AC 41 ms
55,196 KB
testcase_06 AC 688 ms
377,592 KB
testcase_07 AC 321 ms
146,684 KB
testcase_08 AC 651 ms
154,556 KB
testcase_09 AC 535 ms
231,284 KB
testcase_10 AC 464 ms
155,344 KB
testcase_11 AC 359 ms
152,596 KB
testcase_12 AC 357 ms
154,548 KB
testcase_13 AC 498 ms
169,448 KB
testcase_14 AC 470 ms
168,928 KB
testcase_15 AC 1,116 ms
307,208 KB
testcase_16 AC 655 ms
303,160 KB
testcase_17 AC 42 ms
54,644 KB
testcase_18 AC 449 ms
168,744 KB
testcase_19 AC 1,006 ms
244,824 KB
testcase_20 AC 325 ms
154,804 KB
testcase_21 AC 458 ms
171,232 KB
testcase_22 AC 454 ms
171,152 KB
権限があれば一括ダウンロードができます

ソースコード

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