結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー nehan_der_thal
提出日時 2020-08-14 22:52:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 341 ms / 2,000 ms
コード長 1,544 bytes
コンパイル時間 272 ms
コンパイル使用メモリ 82,712 KB
実行使用メモリ 129,112 KB
最終ジャッジ日時 2024-10-10 16:12:05
合計ジャッジ時間 7,165 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFindNode:
    def __init__(self, group_id, parent=None, value=None):
        self.group_id_ = group_id
        self.parent_ = parent
        self.value = value
        self.rank_ = 1
        self.member_num_ = 1

    def is_root(self):
        return not self.parent_

    def root(self):
        parent = self
        while not parent.is_root():
            parent = parent.parent_
            self.parent_ = parent
        return parent

    def find(self):
        root = self.root()
        return root.group_id_

    def rank(self):
        root = self.root()
        return root.rank_

    def unite(self, unite_node):
        root = self.root()
        unite_root = unite_node.root()

        if root.group_id_ != unite_root.group_id_:
            if root.rank() > unite_root.rank():
                n_root, child = root, unite_root
            else:
                n_root, child = unite_root, root
            child.parent_ = n_root
            n_root.rank_ = max(n_root.rank_, child.rank_ + 1)
            n_root.member_num_ = n_root.member_num_ + child.member_num_


N, A, B = map(int, input().split())
X = list(map(int, input().split()))
X.append(10**19)
nodes = [UnionFindNode(i) for i in range(N)]
j=k=-1
for i in range(N):
    while X[j+1] < X[i]+A:
        j+=1
    ks = []
    while X[k+1] <= X[i]+B:
        k+=1
        if j < k:
            ks.append(k)
    if j < k:
        nodes[i].unite(nodes[j+1])
    for kk in ks:
        nodes[i].unite(nodes[kk])
for i in range(N):
    print(nodes[i].root().member_num_)
0