結果
問題 | No.778 クリスマスツリー |
ユーザー | paruki |
提出日時 | 2019-01-22 20:13:59 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 64 ms / 2,000 ms |
コード長 | 1,661 bytes |
コンパイル時間 | 1,545 ms |
コンパイル使用メモリ | 167,920 KB |
実行使用メモリ | 20,988 KB |
最終ジャッジ日時 | 2023-08-04 04:46:31 |
合計ジャッジ時間 | 2,916 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge14 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
8,036 KB |
testcase_01 | AC | 4 ms
8,040 KB |
testcase_02 | AC | 4 ms
8,052 KB |
testcase_03 | AC | 4 ms
8,044 KB |
testcase_04 | AC | 4 ms
8,320 KB |
testcase_05 | AC | 4 ms
8,092 KB |
testcase_06 | AC | 48 ms
20,872 KB |
testcase_07 | AC | 23 ms
9,612 KB |
testcase_08 | AC | 64 ms
14,896 KB |
testcase_09 | AC | 62 ms
11,740 KB |
testcase_10 | AC | 59 ms
11,668 KB |
testcase_11 | AC | 61 ms
11,748 KB |
testcase_12 | AC | 56 ms
11,680 KB |
testcase_13 | AC | 38 ms
11,692 KB |
testcase_14 | AC | 48 ms
20,988 KB |
ソースコード
#define _USE_MATH_DEFINES #include "bits/stdc++.h" using namespace std; #define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i)) #define rep(i,j) FOR(i,0,j) #define each(x,y) for(auto &(x):(y)) #define mp make_pair #define MT make_tuple #define all(x) (x).begin(),(x).end() #define debug(x) cout<<#x<<": "<<(x)<<endl #define smax(x,y) (x)=max((x),(y)) #define smin(x,y) (x)=min((x),(y)) #define MEM(x,y) memset((x),(y),sizeof (x)) #define sz(x) (int)(x).size() #define RT return using ll = long long; using pii = pair<int, int>; using vi = vector<int>; using vll = vector<ll>; template<class Val> struct BinaryIndexedTree { int n; vector<Val> t; BinaryIndexedTree() {} BinaryIndexedTree(int _n) :n(_n + 1), t(_n + 1) {} void add(int k, Val val) { k++; while (k < n) { t[k] += val; k += (k&-k); } } void set(int k, Val val) { add(k, -sum(k, k + 1)); add(k, val); } Val sum(int k) { Val r = 0; while (k > 0) { r += t[k]; k -= (k&-k); } return r; } Val sum(int l, int r) { return sum(r) - sum(l); } }; int N; vi G[200005]; BinaryIndexedTree<int> bit; ll ans; void f(int u) { ans += bit.sum(u); bit.add(u, 1); each(v, G[u]) { f(v); } bit.add(u, -1); } void solve() { cin >> N; rep(i, N - 1) { int p; cin >> p; G[p].push_back(i+1); } bit = BinaryIndexedTree<int>(N); f(0); cout << ans << endl; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout << fixed << setprecision(15); solve(); return 0; }