結果

問題 No.1950 片道きゃっちぼーる
コンテスト
ユーザー terasa
提出日時 2022-05-22 00:46:05
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,582 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 82,392 KB
実行使用メモリ 180,340 KB
最終ジャッジ日時 2024-09-20 12:14:18
合計ジャッジ時間 11,267 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 WA * 6
権限があれば一括ダウンロードができます

ソースコード

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')


class UnionFind:
    def __init__(self, N, X, A):
        self.N = N
        self.par = [-1] * N
        self.m = [x + a for x, a in zip(X, A)]

    def find(self, x):
        if self.par[x] < 0:
            return x
        else:
            self.par[x] = self.find(self.par[x])
            return self.par[x]

    def unite(self, x, y):
        x = self.find(x)
        y = self.find(y)

        if x == y:
            return False
        if x > y:
            x, y = y, x

        self.par[x] += self.par[y]
        self.par[y] = x

        self.m[x] = max(self.m[x], self.m[y])
        return True

    def same(self, x, y):
        return self.find(x) == self.find(y)

    def size(self, x):
        return -self.par[self.find(x)]

    def roots(self):
        return [i for i in range(self.N) if self.par[i] < 0]

    def members(self, x):
        p = self.find(x)
        return [i for i in range(self.N) if self.find(i) == p]


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

uf = UnionFind(N, X, A)
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:
        uf.unite(li, i)
    if ri is not None:
        uf.unite(ri, i)

for i, x in enumerate(X):
    p = uf.find(i)
    print(uf.m[p] - x)
0