結果
問題 | No.778 クリスマスツリー |
ユーザー | fura |
提出日時 | 2020-05-29 00:23:35 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 115 ms / 2,000 ms |
コード長 | 1,265 bytes |
コンパイル時間 | 4,615 ms |
コンパイル使用メモリ | 209,440 KB |
実行使用メモリ | 23,168 KB |
最終ジャッジ日時 | 2024-10-13 15:21:47 |
合計ジャッジ時間 | 5,265 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 1 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 64 ms
23,168 KB |
testcase_07 | AC | 49 ms
17,436 KB |
testcase_08 | AC | 115 ms
19,984 KB |
testcase_09 | AC | 107 ms
16,812 KB |
testcase_10 | AC | 105 ms
16,816 KB |
testcase_11 | AC | 108 ms
16,880 KB |
testcase_12 | AC | 103 ms
16,896 KB |
testcase_13 | AC | 57 ms
16,640 KB |
testcase_14 | AC | 64 ms
23,168 KB |
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i=0;i<(n);i++) using namespace std; using lint=long long; using graph=vector<vector<int>>; void add_edge(graph& G,int u,int v){ G[u].emplace_back(v); G[v].emplace_back(u); } class Euler_tour_for_vertex{ vector<int> L,R; const graph& Tr; int idx; void dfs(int u,int p){ L[u]=idx++; for(int v:Tr[u]) if(v!=p) dfs(v,u); R[u]=idx; } public: Euler_tour_for_vertex(const graph& Tr,int root):L(Tr.size()),R(Tr.size()),Tr(Tr),idx(0){ dfs(root,-1); } int get_index(int u)const{ return L[u]; } pair<int,int> get_subtree(int u)const{ return {L[u],R[u]}; } }; template<class G> class Fenwick_tree_dual{ vector<G> a; public: Fenwick_tree_dual(int n):a(n){} void add(int l,int r,G val){ if(l==0){ for(;r>0;r-=r&-r) a[r-1]+=val; return; } add(0,r,val); add(0,l,-val); } G sum(int i)const{ G res{}; for(i++;i<=a.size();i+=i&-i) res+=a[i-1]; return res; } }; int main(){ int n; scanf("%d",&n); graph T(n); for(int u=1;u<n;u++){ int p; scanf("%d",&p); add_edge(T,p,u); } Euler_tour_for_vertex ET(T,0); Fenwick_tree_dual<int> F(n); lint ans=0; rep(u,n){ int l,r; tie(l,r)=ET.get_subtree(u); ans+=F.sum(l); F.add(l,r,1); } printf("%lld\n",ans); return 0; }