結果

問題 No.2638 Initial fare
コンテスト
ユーザー 👑 ramdos
提出日時 2024-02-11 17:40:23
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 280 bytes
コンパイル時間 519 ms
コンパイル使用メモリ 82,164 KB
実行使用メモリ 296,336 KB
最終ジャッジ日時 2024-09-28 17:38:34
合計ジャッジ時間 4,441 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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)]
v=[list(itertools.chain.from_iterable(e)).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