結果

問題 No.1950 片道きゃっちぼーる
ユーザー terasaterasa
提出日時 2022-05-22 11:00:02
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,172 ms / 3,000 ms
コード長 944 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 81,848 KB
実行使用メモリ 377,320 KB
最終ジャッジ日時 2023-10-20 16:55:10
合計ジャッジ時間 14,396 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
55,848 KB
testcase_01 AC 40 ms
55,848 KB
testcase_02 AC 41 ms
55,848 KB
testcase_03 AC 759 ms
368,588 KB
testcase_04 AC 823 ms
369,996 KB
testcase_05 AC 42 ms
55,848 KB
testcase_06 AC 693 ms
377,320 KB
testcase_07 AC 349 ms
146,096 KB
testcase_08 AC 692 ms
154,016 KB
testcase_09 AC 548 ms
230,764 KB
testcase_10 AC 489 ms
156,888 KB
testcase_11 AC 393 ms
153,984 KB
testcase_12 AC 388 ms
153,984 KB
testcase_13 AC 525 ms
169,140 KB
testcase_14 AC 510 ms
168,692 KB
testcase_15 AC 1,172 ms
308,464 KB
testcase_16 AC 669 ms
303,032 KB
testcase_17 AC 41 ms
55,848 KB
testcase_18 AC 493 ms
168,240 KB
testcase_19 AC 1,092 ms
245,288 KB
testcase_20 AC 374 ms
154,248 KB
testcase_21 AC 489 ms
170,612 KB
testcase_22 AC 490 ms
170,616 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