結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2020-04-14 12:39:46 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 971 ms / 3,000 ms |
コード長 | 650 bytes |
コンパイル時間 | 285 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 113,024 KB |
最終ジャッジ日時 | 2024-10-01 16:55:34 |
合計ジャッジ時間 | 11,331 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
INF = 10 ** 7import sysinput = sys.stdin.readlinesys.setrecursionlimit(100000000)dy = (-1,0,1,0)dx = (0,1,0,-1)MAXN = 200005G = [[] for _ in range(MAXN)]ans = 0def dfs(i,n,p = -1):global ansret = 1for e,c in G[i]:if e == p:continuetmp = dfs(e,n,i)ans += 2 * tmp * (n - tmp) * cret += tmpreturn retdef main():n = int(input())for _ in range(n - 1):a,b,c = map(int,input().split())a -= 1b -= 1G[a].append((b,c))G[b].append((a,c))dfs(0,n)global ansprint(ans)if __name__ == '__main__':main()