結果

問題 No.1843 Tree ANDistance
ユーザー ああいい
提出日時 2022-02-18 22:03:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,095 ms / 2,000 ms
コード長 1,030 bytes
コンパイル時間 162 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 141,056 KB
最終ジャッジ日時 2024-06-29 08:58:35
合計ジャッジ時間 17,946 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
r = sys.stdin
sys.setrecursionlimit(10 ** 8)
N = int(r.readline())
_max = 0
G = [[] for _ in range(N+1)]
for _ in range(N-1):
    a,b,c = map(int,r.readline().split())
    if c > _max:
        _max = c
    G[a].append((b,c))
    G[b].append((a,c))
k = 0
while 1 << k < _max:
    k += 1
ans = 0
P = 10 ** 9 + 7
def find(i):
    if parent[i] < 0:return i
    parent[i] = find(parent[i])
    return parent[i]
def unite(i,j):
    I = find(i)
    J = find(j)
    if I == J:return False
    if parent[I] < parent[J]:
        I,J = J,I
    parent[J] += parent[I]
    parent[I] = J
    return True
for i in range(k+1):
    mask = 1 << i
    parent = [-1] * (N+1)
    stack = [(1,-1)]
    while stack:
        now,p = stack.pop()
        for v,c in G[now]:
            if v == p:continue
            if c & mask:
                unite(now,v)
            stack.append((v,now))
    for i in range(1,N+1):
        if parent[i] < 0:
            num = -parent[i]
            ans = (ans + num * (num-1) //2 % P * mask) % P
print(ans)
0