結果
問題 | 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 sysr = sys.stdinsys.setrecursionlimit(10 ** 8)N = int(r.readline())_max = 0G = [[] for _ in range(N+1)]for _ in range(N-1):a,b,c = map(int,r.readline().split())if c > _max:_max = cG[a].append((b,c))G[b].append((a,c))k = 0while 1 << k < _max:k += 1ans = 0P = 10 ** 9 + 7def find(i):if parent[i] < 0:return iparent[i] = find(parent[i])return parent[i]def unite(i,j):I = find(i)J = find(j)if I == J:return Falseif parent[I] < parent[J]:I,J = J,Iparent[J] += parent[I]parent[I] = Jreturn Truefor i in range(k+1):mask = 1 << iparent = [-1] * (N+1)stack = [(1,-1)]while stack:now,p = stack.pop()for v,c in G[now]:if v == p:continueif 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) % Pprint(ans)