結果
問題 | No.778 クリスマスツリー |
ユーザー |
![]() |
提出日時 | 2021-11-07 18:48:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 128 ms / 2,000 ms |
コード長 | 2,125 bytes |
コンパイル時間 | 3,256 ms |
コンパイル使用メモリ | 175,748 KB |
最終ジャッジ日時 | 2025-01-25 14:41:40 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
#include <algorithm>#include <iomanip>#include <array>#include <bitset>#include <cassert>#include <cmath>#include <cstdio>#include <deque>#include <functional>#include <iostream>#include <iterator>#include <map>#include <queue>#include <set>#include <string>#include <sstream>#include <unordered_map>#include <unordered_set>#include <utility>#include <numeric>#include <vector>#include <climits>using namespace std;#if __has_include(<atcoder/all>)#include <atcoder/all>using namespace atcoder;#endif#define GET_MACRO(_1, _2, _3, NAME, ...) NAME#define _rep(i, n) _rep2(i, 0, n)#define _rep2(i, a, b) for(int i = (int)(a); i < (int)(b); i++)#define rep(...) GET_MACRO(__VA_ARGS__, _rep2, _rep)(__VA_ARGS__)#define all(x) (x).begin(), (x).end()#define rall(x) (x).rbegin(), (x).rend()using i64 = long long;template<class T, class U>bool chmin(T& a, const U& b) { return (b < a) ? (a = b, true) : false; }template<class T, class U>bool chmax(T& a, const U& b) { return (b > a) ? (a = b, true) : false; }template<typename T>istream& operator>>(istream&i,vector<T>&v){rep(j,v.size())i>>v[j];return i;}template<typename T>string join(vector<T>&v){stringstream s;rep(i,v.size())s<<' '<<v[i];return s.str().substr(1);}template<typename T>ostream& operator<<(ostream&o,vector<T>&v){if(v.size())o<<join(v);return o;}template<typename T>string join(vector<vector<T>>&vv){string s="\n";rep(i,vv.size())s+=join(vv[i])+"\n";return s;}template<typename T>ostream& operator<<(ostream&o,vector<vector<T>>&vv){if(vv.size())o<<join(vv);return o;}template<class T> using pq = priority_queue<T, vector<T>, greater<T>>;i64 ans = 0;vector<vector<int>> edge;void dfs(int now, int p, fenwick_tree<i64>& fw) {fw.add(now, 1);ans += fw.sum(0, now);for(auto i : edge[now]) {if(i != p) dfs(i, now, fw);}fw.add(now, -1);}int main() {int n;cin >> n;fenwick_tree<i64> fw(n + 1);edge.resize(n);rep(i, n - 1) {int a;cin >> a;edge[a].push_back(i + 1);}dfs(0, -1, fw);cout << ans << endl;return 0;}