結果

問題 No.2638 Initial fare
ユーザー 👑 ramdos
提出日時 2024-02-11 17:41:20
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 284 bytes
コンパイル時間 483 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 123,688 KB
最終ジャッジ日時 2024-09-28 17:38:39
合計ジャッジ時間 4,241 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 1 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

import itertools
N=int(input())
e=[list(map(lambda x: int(x)-1,input().split())) for i in range(N-1)]
q=list(itertools.chain.from_iterable(e))
v=[q.count(i) for i in range(N)]
print(N-1+sum([v[i]*(v[i]-1)//2 for i in range(N)])+sum([(v[e[i][0]]-1)*(v[e[i][1]]-1)for i in range(N-1)]))
0