結果

問題 No.872 All Tree Path
ユーザー AT274_AT274_
提出日時 2019-10-14 15:34:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 691 ms / 3,000 ms
コード長 1,356 bytes
コンパイル時間 892 ms
コンパイル使用メモリ 81,664 KB
実行使用メモリ 155,776 KB
最終ジャッジ日時 2024-12-24 01:34:24
合計ジャッジ時間 8,568 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 675 ms
155,776 KB
testcase_01 AC 689 ms
152,832 KB
testcase_02 AC 653 ms
143,872 KB
testcase_03 AC 371 ms
142,488 KB
testcase_04 AC 42 ms
51,968 KB
testcase_05 AC 656 ms
143,872 KB
testcase_06 AC 686 ms
155,776 KB
testcase_07 AC 691 ms
153,472 KB
testcase_08 AC 148 ms
81,792 KB
testcase_09 AC 147 ms
82,176 KB
testcase_10 AC 148 ms
81,792 KB
testcase_11 AC 147 ms
81,664 KB
testcase_12 AC 145 ms
81,792 KB
testcase_13 AC 43 ms
52,096 KB
testcase_14 AC 44 ms
51,712 KB
testcase_15 AC 45 ms
51,840 KB
testcase_16 AC 45 ms
52,608 KB
testcase_17 AC 44 ms
52,096 KB
testcase_18 AC 45 ms
52,224 KB
testcase_19 AC 45 ms
52,224 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# nadare様のコード参考
# https://yukicoder.me/submissions/374115

N = int(input())
T = [[] for i in range(N)]
E = []
for i in range(N - 1):
    a, b, w = map(int, input().split())
    a, b = a - 1, b - 1
    T[a].append(b)
    T[b].append(a)
    E.append([a, b, w])


class TwoDividedTree:
    def __init__(self, n, tree):
        self.tree = tree
        self.depth = [-1] * n
        self.size = [1] * n

        self.calc()

    def calc(self):
        self.depth[0] = 0
        stack = [0]
        rev_directions = []  # 深いところから根に辿る辺の集合

        while stack:
            fr = stack.pop()
            for to in self.tree[fr]:
                if self.depth[to] != -1:
                    continue

                self.depth[to] = self.depth[fr] + 1
                stack.append(to)
                rev_directions.append([fr, to])

        while rev_directions:
            parent, child = rev_directions.pop()
            self.size[parent] += self.size[child]

    '''
    辺（a, b） で木を２分割した時の、片方のサイズ
    '''
    def size_after_divided(self, a, b):
        if self.depth[a] < self.depth[b]:
            a, b = b, a
        return self.size[a]


tdt = TwoDividedTree(N, T)
ans = 0
for a, b, w in E:
    s = tdt.size_after_divided(a, b)
    ans += 2 * s * (N - s) * w

print(ans)
0