結果
問題 | No.778 クリスマスツリー |
ユーザー | fura |
提出日時 | 2020-05-29 00:23:35 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 132 ms / 2,000 ms |
コード長 | 1,265 bytes |
コンパイル時間 | 2,981 ms |
コンパイル使用メモリ | 203,272 KB |
実行使用メモリ | 23,176 KB |
最終ジャッジ日時 | 2024-04-21 16:25:51 |
合計ジャッジ時間 | 4,501 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,812 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,940 KB |
testcase_03 | AC | 2 ms
6,944 KB |
testcase_04 | AC | 2 ms
6,940 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 62 ms
23,168 KB |
testcase_07 | AC | 45 ms
17,436 KB |
testcase_08 | AC | 132 ms
19,984 KB |
testcase_09 | AC | 122 ms
16,760 KB |
testcase_10 | AC | 121 ms
16,696 KB |
testcase_11 | AC | 123 ms
16,996 KB |
testcase_12 | AC | 116 ms
16,940 KB |
testcase_13 | AC | 57 ms
16,744 KB |
testcase_14 | AC | 62 ms
23,176 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; }