結果

問題 No.2638 Initial fare
ユーザー flippergo
提出日時 2025-02-22 18:55:14
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 557 bytes
コンパイル時間 711 ms
コンパイル使用メモリ 82,184 KB
実行使用メモリ 121,196 KB
最終ジャッジ日時 2025-02-22 18:55:22
合計ジャッジ時間 7,521 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 TLE * 1 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
G = {i:[] for i in range(1,N+1)}
for _ in range(N-1):
    u,v = map(int,input().split())
    G[u].append(v)
    G[v].append(u)
def f(i,pi,n):
    if n==1:
        cnt = 0
        for j in G[i]:
            if j==pi:continue
            cnt += 1
        return cnt+1
    cnt = 0
    for j in G[i]:
        if j==pi:continue
        cnt += f(j,i,n-1)
    if n==2:
        return cnt+1
    return cnt
ans = 0
def dfs(i,pi):
    global ans
    ans += f(i,0,3)
    for j in G[i]:
        if j==pi:continue
        dfs(j,i)
dfs(1,0)
print(ans//2)
0