結果
問題 | 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 |
ソースコード
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)