結果
問題 | No.778 クリスマスツリー |
ユーザー |
![]() |
提出日時 | 2021-02-10 11:50:58 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,334 ms / 2,000 ms |
コード長 | 1,355 bytes |
コンパイル時間 | 193 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 81,416 KB |
最終ジャッジ日時 | 2024-07-07 22:10:43 |
合計ジャッジ時間 | 11,355 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
import syssys.setrecursionlimit(10**6)int1 = lambda x: int(x)-1p2D = lambda x: print(*x, sep="\n")def II(): return int(sys.stdin.buffer.readline())def MI(): return map(int, sys.stdin.buffer.readline().split())def MI1(): return map(int1, sys.stdin.buffer.readline().split())def LI(): return list(map(int, sys.stdin.buffer.readline().split()))def LI1(): return list(map(int1, sys.stdin.buffer.readline().split()))def LLI(rows_number): return [LI() for _ in range(rows_number)]def BI(): return sys.stdin.buffer.readline().rstrip()def SI(): return sys.stdin.buffer.readline().rstrip().decode()# dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]dij = [[(-1, 0), (1, 0)], [(0, -1), (0, 1)]]inf = 10**16md = 998244353# md = 10**9+7class BitSum:def __init__(self, n):self.n = n+1self.table = [0]*self.ndef add(self, i, x):i += 1while i < self.n:self.table[i] += xi += i & -i# [0,i]の和def sum(self, i):i += 1res = 0while i > 0:res += self.table[i]i -= i & -ireturn resn=II()to=[[] for _ in range(n)]for c,p in enumerate(MI(),1):to[p].append(c)bit=BitSum(n)def dfs(u):res=bit.sum(u)bit.add(u,1)for v in to[u]:res+=dfs(v)bit.add(u,-1)return resprint(dfs(0))