結果
問題 | No.778 クリスマスツリー |
ユーザー |
![]() |
提出日時 | 2020-12-03 10:56:11 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,657 ms / 2,000 ms |
コード長 | 742 bytes |
コンパイル時間 | 279 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 79,520 KB |
最終ジャッジ日時 | 2024-09-13 18:50:55 |
合計ジャッジ時間 | 14,270 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
import sysinput = sys.stdin.readlinesys.setrecursionlimit(2*10**5+10)class BIT:def __init__(self, n):self.n = nself.bit = [0]*(n+1)def add(self, i, x):i += 1while i<=self.n:self.bit[i] += xi += i&(-i)def acc(self, i):s = 0while i>0:s += self.bit[i]i -= i&(-i)return sdef dfs(v):global ansans += bit.acc(v)bit.add(v, 1)for nv in G[v]:dfs(nv)bit.add(v, -1)N = int(input())A = list(map(int, input().split()))G = [[] for _ in range(N)]for i in range(N-1):G[A[i]].append(i+1)bit = BIT(N)ans = 0dfs(0)print(ans)