結果

問題 No.2638 Initial fare
ユーザー katonyonkokatonyonko
提出日時 2024-02-19 21:45:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 935 ms / 2,000 ms
コード長 1,930 bytes
コンパイル時間 139 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 141,144 KB
最終ジャッジ日時 2024-02-19 21:46:16
合計ジャッジ時間 15,404 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge16
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
55,728 KB
testcase_01 AC 41 ms
55,728 KB
testcase_02 AC 42 ms
55,728 KB
testcase_03 AC 36 ms
55,728 KB
testcase_04 AC 739 ms
131,492 KB
testcase_05 AC 771 ms
135,284 KB
testcase_06 AC 37 ms
55,728 KB
testcase_07 AC 675 ms
137,076 KB
testcase_08 AC 366 ms
130,704 KB
testcase_09 AC 529 ms
141,144 KB
testcase_10 AC 426 ms
140,852 KB
testcase_11 AC 36 ms
55,728 KB
testcase_12 AC 721 ms
135,624 KB
testcase_13 AC 487 ms
128,960 KB
testcase_14 AC 649 ms
139,852 KB
testcase_15 AC 565 ms
140,584 KB
testcase_16 AC 38 ms
55,728 KB
testcase_17 AC 781 ms
134,972 KB
testcase_18 AC 580 ms
135,444 KB
testcase_19 AC 882 ms
134,964 KB
testcase_20 AC 863 ms
133,680 KB
testcase_21 AC 893 ms
133,420 KB
testcase_22 AC 39 ms
55,728 KB
testcase_23 AC 863 ms
130,756 KB
testcase_24 AC 930 ms
133,364 KB
testcase_25 AC 38 ms
55,728 KB
testcase_26 AC 935 ms
134,132 KB
testcase_27 AC 931 ms
133,488 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import io
import sys
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, accumulate
from heapq import heappush, heappop
from bisect import bisect_right, bisect_left
from math import gcd
import math

_INPUT = """\
6
5
1 2
2 3
3 4
4 5
8
1 2
2 3
2 4
2 5
5 6
6 7
6 8
2
1 2
"""

def bfs(G,s):
  inf=10**30
  D=[inf]*len(G)
  D[s]=0
  dq=deque()
  dq.append(s)
  while dq:
    x=dq.popleft()
    for y in G[x]:
      if D[y]>D[x]+1:
        D[y]=D[x]+1
        dq.append(y)
  return D

def input():
  return sys.stdin.readline()[:-1]

def solve(test):
  N=int(input())
  G=[[] for _ in range(N)]
  for _ in range(N-1):
    a,b=map(int, input().split())
    a-=1
    b-=1
    G[a].append(b)
    G[b].append(a)
  D=bfs(G,0)
  order=sorted([(D[i],i) for i in range(N)],reverse=True)
  ans=0
  childs=[[0]*3 for _ in range(N)]
  for d,i in order:
    tmp=0
    for c in G[i]:
      if D[c]>d:
        childs[i][1]+=childs[c][0]
        childs[i][2]+=childs[c][1]
        tmp+=1
    childs[i][0]=tmp
    ans+=sum(childs[i])
    ans+=childs[i][0]*(childs[i][0]-1)//2
    for c in G[i]:
      if D[c]>d:
        ans+=childs[i][1]-childs[c][0]
  print(ans)

def random_input():
  from random import randint,shuffle
  N=randint(1,10)
  M=randint(1,N)
  A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]
  shuffle(A)
  return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3

def simple_solve():
  return []

def main(test):
  if test==0:
    solve(0)
  elif test==1:
    sys.stdin = io.StringIO(_INPUT)
    case_no=int(input())
    for _ in range(case_no):
      solve(0)
  else:
    for i in range(1000):
      sys.stdin = io.StringIO(random_input())
      x=solve(1)
      y=simple_solve()
      if x!=y:
        print(i,x,y)
        print(*[line for line in sys.stdin],sep='')
        break

#0:提出用、1:与えられたテスト用、2:ストレステスト用
main(0)
0