結果
問題 | No.778 クリスマスツリー |
ユーザー |
|
提出日時 | 2021-02-01 12:51:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 704 ms / 2,000 ms |
コード長 | 1,582 bytes |
コンパイル時間 | 172 ms |
コンパイル使用メモリ | 82,656 KB |
実行使用メモリ | 379,972 KB |
最終ジャッジ日時 | 2024-06-29 22:48:00 |
合計ジャッジ時間 | 6,574 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#region Header#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]sys.setrecursionlimit(1000000)#endregion# _INPUT = """3# 2 0# """# sys.stdin = io.StringIO(_INPUT)# 最もシンプルclass BIT:"""Binary Indexed Tree (Fenwick Tree), 1-indexed"""def __init__(self, n):"""Parameters----------n : int要素数。index は 0..n になる。"""self.size = nself.data = [0] * (n+1)# self.depth = n.bit_length()def add(self, i, x):while i <= self.size:self.data[i] += xi += i & -idef get_sum(self, i):s = 0while i > 0:s += self.data[i]i -= i & -ireturn sdef get_rsum(self, l, r):"""[l, r) の sum"""return self.get_sum(r) - self.get_sum(l-1)def dfs(G, bit, N, pos):n = bit.get_sum(pos+1)bit.add(pos+1, 1)for next_pos in G[pos]:n += dfs(G, bit, N, next_pos)bit.add(pos+1, -1)return ndef main():N = int(input())A = list(map(int, input().split()))G = [list() for _ in range(N)]for i in range(N-1):G[A[i]].append(i+1)bit = BIT(N)ans = dfs(G, bit, N, 0)print(ans)if __name__ == '__main__':main()