結果
問題 | No.778 クリスマスツリー |
ユーザー | edamame882 |
提出日時 | 2019-06-24 09:34:04 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 73 ms / 2,000 ms |
コード長 | 1,748 bytes |
コンパイル時間 | 1,561 ms |
コンパイル使用メモリ | 174,300 KB |
実行使用メモリ | 28,160 KB |
最終ジャッジ日時 | 2024-10-14 02:00:18 |
合計ジャッジ時間 | 2,845 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 1 ms
5,248 KB |
testcase_05 | AC | 2 ms
5,248 KB |
testcase_06 | AC | 57 ms
28,160 KB |
testcase_07 | AC | 24 ms
10,432 KB |
testcase_08 | AC | 72 ms
18,816 KB |
testcase_09 | AC | 71 ms
12,544 KB |
testcase_10 | AC | 70 ms
12,544 KB |
testcase_11 | AC | 69 ms
12,672 KB |
testcase_12 | AC | 73 ms
12,672 KB |
testcase_13 | AC | 41 ms
12,672 KB |
testcase_14 | AC | 56 ms
28,160 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; //repetition #define FOR(i,a,b) for(ll i=(a);i<(b);++i) #define rep(i, n) for(ll i = 0; i < (ll)(n); i++) //container util #define all(x) (x).begin(),(x).end() //typedef typedef long long ll; typedef vector<int> VI; typedef vector<VI> VVI; typedef vector<ll> VLL; typedef vector<VLL> VVLL; typedef vector<string> VS; typedef pair<int, int> PII; typedef pair<ll, ll> PLL; //const value //const ll MOD = 1e9 + 7; //const int dx[] = {0,1,0,-1};//{0,0,1,1,1,-1,-1,-1}; //const int dy[] = {1,0,-1,0};//{1,-1,0,1,-1,0,1,-1}; //conversion inline int toInt(string s) {int v; istringstream sin(s);sin>>v;return v;} inline ll toLL(string s) {ll v; istringstream sin(s);sin>>v;return v;} template<class T> inline string toString(T x) {ostringstream sout;sout<<x;return sout.str();} struct BIT{ int n; VI nodes; void init(int N){ nodes.resize(N+1,0); n = N; } void add(int i, int x){ for(int idx = i; idx <= n; idx += idx & (-idx)){ nodes[idx] += x; } } ll sum(int x){ ll res = 0; for(int idx = x; idx > 0 ; idx -= idx & (-idx)){ res += nodes[idx]; } return res; } }; ll dfs(int current_node, VVI& edge, BIT & bit){ ll res = 0; // 二分探索をしてcurrent_nodeよりも小さい要素を探索 res += bit.sum(current_node+1); bit.add(current_node+1,1); for(int node :edge[current_node]){ res += dfs(node,edge,bit); } bit.add(current_node+1,-1); return res; } int main(){ ios::sync_with_stdio(false); cin.tie(0); int n; cin >> n; VI a(n); VVI edge(n); FOR(i,1,n){ cin >> a[i]; edge[a[i]].push_back(i); } BIT bit; bit.init(n); ll ans = dfs(0,edge,bit); cout << ans << endl; return 0; }