結果
問題 |
No.778 クリスマスツリー
|
ユーザー |
![]() |
提出日時 | 2025-01-31 04:41:50 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,589 bytes |
コンパイル時間 | 747 ms |
コンパイル使用メモリ | 12,160 KB |
実行使用メモリ | 63,660 KB |
最終ジャッジ日時 | 2025-01-31 04:42:04 |
合計ジャッジ時間 | 13,008 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 9 RE * 3 |
ソースコード
import sys input = sys.stdin.readline class Bit_indexed_tree(): def __init__(self, LEN): self.BIT = [0]*(LEN+1) # 1-indexedなtree. 配列BITの長さはLEN+1にしていることに注意。 self.LEN = LEN def update(self,v,w): # index vにwを加える while v<=self.LEN: self.BIT[v]+=w v+=(v&(-v)) # v&(-v)で、最も下の立っているビット. 自分を含む大きなノードへ. たとえばv=3→v=4 def getvalue(self,v): # [1,v]の区間の和を求める ANS=0 while v!=0: ANS+=self.BIT[v] v-=(v&(-v)) # 自分より小さい自分の和を構成するノードへ. たとえばv=14→v=12へ return ANS def bisect_on_BIT(self,x): # [1,ind]の和がはじめてx以上になるindexを探す if x<=0: return 0 ANS=0 h=1<<((self.LEN).bit_length()-1) # LEN以下の最小の2ベキ while h>0: if ANS+h<=self.LEN and self.BIT[ANS+h]<x: x-=self.BIT[ANS+h] ANS+=h h//=2 return ANS+1 # LENまでの和がx未満のとき, LEN+1を返すことに注意 N=int(input()) A=list(map(int,input().split())) E=[[] for i in range(N)] Parent=[-1]*N Child=[[] for i in range(N)] for i in range(1,N): x=A[i-1] Parent[i]=x Child[x].append(i) BIT=Bit_indexed_tree(N+10) ANS=[0] def dfs(x): ANS[0]+=BIT.getvalue(x+1) BIT.update(x+1,1) for c in Child[x]: dfs(c) BIT.update(x+1,-1) dfs(0) print(ANS[0])