結果
問題 | No.778 クリスマスツリー |
ユーザー |
|
提出日時 | 2019-01-22 20:13:59 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 72 ms / 2,000 ms |
コード長 | 1,661 bytes |
コンパイル時間 | 1,569 ms |
コンパイル使用メモリ | 170,124 KB |
実行使用メモリ | 21,200 KB |
最終ジャッジ日時 | 2024-10-14 01:56:48 |
合計ジャッジ時間 | 2,687 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#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; }