結果
問題 | No.778 クリスマスツリー |
ユーザー |
![]() |
提出日時 | 2019-01-01 15:39:34 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 93 ms / 2,000 ms |
コード長 | 2,593 bytes |
コンパイル時間 | 1,549 ms |
コンパイル使用メモリ | 170,732 KB |
実行使用メモリ | 24,640 KB |
最終ジャッジ日時 | 2024-10-14 01:54:54 |
合計ジャッジ時間 | 3,229 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#include<bits/stdc++.h>#define rep(i,a,b) for(int i=a;i<b;i++)#define rrep(i,a,b) for(int i=a;i>=b;i--)#define fore(i,a) for(auto &i:a)#define all(x) (x).begin(),(x).end()#pragma GCC optimize ("-O3")using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); }typedef long long ll; const int inf = INT_MAX / 2; const ll infl = 1LL << 60;template<class T>bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }template<class T>bool chmin(T &a, const T &b) { if (b<a) { a = b; return 1; } return 0; }//---------------------------------------------------------------------------------------------------template<class V> struct BIT {BIT() {} // [L, R)int NV;vector<V> bit;BIT(int n){ init(n); }void init(int n) { NV = 1; while (NV < n)NV *= 2; bit.resize(NV); clear(); }V operator[](int e) { V s = 0; e++; while (e) s += bit[e - 1], e -= e&-e; return s; }void add(int e, V v) { e++; while (e <= NV) bit[e - 1] += v, e += e&-e; }int lower_bound(V val) {V tv = 0; int i, ent = 0; for (i = NV - 1; i >= 0; i--)if(tv+bit[ent+(1<<i)-1]<=val)tv+=bit[ent+(1<<i)-1],ent += (1 << i);return ent;}V get(int L, int R) {assert(0 <= L); assert(R <= NV); assert(L <= R);V res = 0; if(R) res += operator[](R - 1); if (L) res -= operator[](L - 1);return res;}void clear() { rep(i, 0, NV) bit[i] = 0; }};/*---------------------------------------------------------------------------------------------------∧_∧∧_∧ (´<_` ) Welcome to My Coding Space!( ´_ゝ`) / ⌒i/ \ | |/ / ̄ ̄ ̄ ̄/ |__(__ニつ/ _/ .| .|____\/____/ (u ⊃---------------------------------------------------------------------------------------------------*/int N;vector<int> E[201010];BIT<int> bit(201010);//---------------------------------------------------------------------------------------------------ll ans = 0;void dfs(int cu, int pa = -1) {ans += bit.get(0, cu);bit.add(cu, 1);fore(to, E[cu]) if (to != pa) dfs(to, cu);bit.add(cu, -1);}//---------------------------------------------------------------------------------------------------void _main() {cin >> N;rep(i, 1, N) {int a; cin >> a;E[i].push_back(a);E[a].push_back(i);}dfs(0);cout << ans << endl;}