結果

問題 No.386 貪欲な領主
ユーザー rpy3cpprpy3cpp
提出日時 2016-07-01 23:24:05
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 3,065 bytes
コンパイル時間 145 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 131,112 KB
最終ジャッジ日時 2024-10-12 19:06:20
合計ジャッジ時間 11,800 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 30 ms
11,008 KB
testcase_01 AC 32 ms
11,008 KB
testcase_02 AC 34 ms
11,008 KB
testcase_03 AC 31 ms
11,136 KB
testcase_04 RE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 AC 50 ms
12,544 KB
testcase_08 AC 287 ms
30,952 KB
testcase_09 AC 59 ms
13,292 KB
testcase_10 AC 30 ms
11,136 KB
testcase_11 AC 29 ms
11,008 KB
testcase_12 AC 40 ms
11,648 KB
testcase_13 AC 80 ms
14,720 KB
testcase_14 TLE -
testcase_15 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

def read_data():
    N = int(input())
    Es = [[] for i in range(N)]
    for i in range(N - 1):
        a, b = map(int, input().split())
        Es[a].append(b)
        Es[b].append(a)
    Us = [int(input()) for i in range(N)]
    M = int(input())
    moves = [list(map(int, input().split())) for m in range(M)]
    return N, Es, Us, M, moves


class DisjointSet(object):
    def __init__(self, n):
        self.parent = list(range(n))
        self.rank = [0] * n
    
    def union(self, x, y):
        self._link(self.find_set(x), self.find_set(y))
    
    def _link(self, x, y):
        if self.rank[x] > self.rank[y]:
            self.parent[y] = x
        else:
            self.parent[x] = y
            if self.rank[x] == self.rank[y]:
                self.rank[y] += 1
    
    def find_set(self, x):
        xp = self.parent[x]
        if xp != x:
            self.parent[x] = self.find_set(xp)
        return self.parent[x]

class Tree(object):
    def __init__(self, N, Es, root, Us):
        self.n = N
        self.root = root
        self.child = [[] for i in range(N)]
        self.cum_cost = [0 for i in range(N)]
        self.cost = Us
        self._set_child(Es)
    
    def _set_child(self, Es):
        que = [self.root]
        visited = [False] * self.n
        self.cum_cost[self.root] = self.cost[self.root]
        while que:
            v = que.pop()
            cum_cost_v = self.cum_cost[v]
            for u in Es[v]:
                if visited[u]:
                    continue
                self.child[v].append(u)
                self.cum_cost[u] = cum_cost_v + self.cost[u]
                que.append(u)
            visited[v] = True        

class LCATarjan(object):
    def __init__(self, tree):
        self.n = tree.n
        self.root = tree.root
        self.child = tree.child
        self.ancestor = list(range(self.n))
        self.visited = [False] * self.n
        self.ds = DisjointSet(self.n)
        self.lca = dict()
    
    def find(self, pairs):
        self._preprocess(pairs)
        self._LCA(self.root)
        return self.lca

    def _preprocess(self, pairs):
        self.pairs = [[] for node in range(self.n)]
        for u, v in pairs:
            self.pairs[u].append(v)
            self.pairs[v].append(u)
    
    def _LCA(self, u):
        self.ancestor[self.ds.find_set(u)] = u
        for v in self.child[u]:
            self._LCA(v)
            self.ds.union(u, v)
            self.ancestor[self.ds.find_set(u)] = u
        self.visited[u] = True
        for v in self.pairs[u]:
            if self.visited[v]:
                self.lca[u, v] = self.ancestor[self.ds.find_set(v)]
                self.lca[v, u] = self.lca[u, v]

    

def solve(N, Es, Us, M, moves):
    tree = Tree(N, Es, 0, Us)
    pairs = [(a, b) for a, b, c in moves]
    lcat = LCATarjan(tree)
    lca = lcat.find(pairs)
    tax = 0
    for a, b, c in moves:
        v = lca[a, b]
        tax += (tree.cum_cost[a] + tree.cum_cost[b] - tree.cum_cost[v] * 2 + Us[v]) * c
    return tax

pars = read_data()
print(solve(*pars))
0