結果

問題 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)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0