結果
問題 | No.1843 Tree ANDistance |
ユーザー |
|
提出日時 | 2022-02-19 16:34:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 495 ms / 2,000 ms |
コード長 | 1,088 bytes |
コンパイル時間 | 236 ms |
コンパイル使用メモリ | 82,280 KB |
実行使用メモリ | 113,868 KB |
最終ジャッジ日時 | 2024-06-29 10:27:52 |
合計ジャッジ時間 | 12,263 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 38 |
ソースコード
from ctypes import Unionclass Unionfind:def __init__(self,n):self.uf = [-1]*ndef find(self,x):if self.uf[x] < 0:return xelse:self.uf[x] = self.find(self.uf[x])return self.uf[x]def same(self,x,y):return self.find(x) == self.find(y)def union(self,x,y):x = self.find(x)y = self.find(y)if x == y:return Falseif self.uf[x] > self.uf[y]:x,y = y,xself.uf[x] += self.uf[y]self.uf[y] = xreturn Truedef size(self,x):x = self.find(x)return -self.uf[x]n = int(input())ans = 0mod = 10**9+7edges = [list(map(int,input().split())) for i in range(n-1)]for i in range(30):uf = Unionfind(n)for a,b,c in edges:if c >> i & 1:uf.union(a-1,b-1)count = 0for ind in range(n):if uf.find(ind) != ind:continues = uf.size(ind)count += s*(s-1)//2ans += (count<<i)%modans %= modprint(ans)