結果
| 問題 | 
                            No.2638 Initial fare
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2024-02-19 21:58:38 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 526 ms / 2,000 ms | 
| コード長 | 558 bytes | 
| コンパイル時間 | 262 ms | 
| コンパイル使用メモリ | 82,408 KB | 
| 実行使用メモリ | 131,848 KB | 
| 最終ジャッジ日時 | 2024-09-29 01:57:18 | 
| 合計ジャッジ時間 | 11,382 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge2 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 25 | 
ソースコード
import collections,sys,math,functools,operator,itertools,bisect,heapq,decimal,string,time,random
#sys.setrecursionlimit(10**9)
#sys.set_int_max_str_digits(0)
n = int(input())
edge = [[] for i in range(n)]
nyuji = [0 for i in range(n)]
e = []
for i in range(n-1):
    u,v = map(int,input().split())
    u-=1
    v-=1
    edge[u].append(v)
    edge[v].append(u)
    nyuji[u] += 1
    nyuji[v] += 1
    e.append((u,v))
ans = 0
for u,v in e:
    ans += (nyuji[u]-1) * (nyuji[v] - 1 )
    ans += 1
for i in range(n):
    ans += nyuji[i]*(nyuji[i]-1)//2
print(ans)