結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2020-02-20 22:04:47 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 780 ms / 3,000 ms |
コード長 | 961 bytes |
コンパイル時間 | 371 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 164,632 KB |
最終ジャッジ日時 | 2024-10-08 19:17:49 |
合計ジャッジ時間 | 9,465 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
def main():import sysfrom collections import dequeinput = sys.stdin.readlineN = int(input())adj = [[] for _ in range(N+1)]E = {}for _ in range(N-1):a, b, w = map(int, input().split())adj[a].append(b)adj[b].append(a)E[a*(N+1)+b] = wE[b*(N+1)+a] = wque = deque()que.append(1)seen = [-1] * (N+1)seen[1] = 0par = [0] * (N+1)child = [[] for _ in range(N+1)]seq = []while que:v = que.popleft()seq.append(v)for u in adj[v]:if seen[u] == -1:seen[u] = seen[v] + 1par[u] = vchild[v].append(u)que.append(u)seq.reverse()dp = [1] * (N+1)ans = 0for v in seq:for u in child[v]:dp[v] += dp[u]ans += E[v*(N+1)+u] * dp[u] * (N-dp[u]) * 2print(ans)if __name__ == '__main__':main()