結果
問題 | No.778 クリスマスツリー |
ユーザー |
|
提出日時 | 2019-03-09 20:55:49 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 72 ms / 2,000 ms |
コード長 | 2,195 bytes |
コンパイル時間 | 849 ms |
コンパイル使用メモリ | 105,356 KB |
実行使用メモリ | 29,860 KB |
最終ジャッジ日時 | 2024-10-14 01:58:35 |
合計ジャッジ時間 | 2,028 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
// includes#include <cstdio>#include <cstdint>#include <iostream>#include <iomanip>#include <string>#include <queue>#include <stack>#include <vector>#include <set>#include <map>#include <unordered_map>#include <algorithm>#include <utility>#include <functional>#include <cmath>#include <climits>#include <bitset>#include <list>#include <random>// macros#define ll long long int#define pb emplace_back#define mk make_pair#define pq priority_queue#define FOR(i, a, b) for(int i=(a);i<(b);++i)#define rep(i, n) FOR(i, 0, n)#define rrep(i, n) for(int i=((int)(n)-1);i>=0;i--)#define all(x) (x).begin(),(x).end()#define sz(x) ((int)(x).size())#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end())using namespace std;// typestypedef pair<int, int> P;typedef pair<ll, int> Pl;typedef pair<ll, ll> Pll;typedef pair<double, double> Pd;// constantsconst int inf = 1e9;const ll linf = 1LL << 50;const double EPS = 1e-10;const int mod = 1e9 + 7;// solvetemplate <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(a > b){a = b; return 1;} return 0;}vector<int> child[200001];vector<int> eu;int cs[200001];template<typename T>struct BIT_{int n;vector<T> bit;BIT_(int n_){n = n_;bit = vector<T>(n+1, 0);}void add(int a, T w){for(int x = a; x < n; x |= (x + 1)){bit[x] += w;}}T sum(int a){T res = 0;for(int x = a; x >= 0; x = (x & (x + 1)) - 1){res += bit[x];}return res;}};typedef struct BIT_<int> BITI;typedef struct BIT_<long long> BITL;int dfs(int i){int res = 0;eu.pb(i);for(auto c: child[i]){res += dfs(c);}cs[i] = res;return res + 1;}int main(int argc, char const* argv[]){ios_base::sync_with_stdio(false);cin.tie(0);int n;cin >> n;rep(i, n - 1){int x;cin >> x;child[x].pb(i+1);}dfs(0);vector<int> tr(n, 0);rep(i, n)tr[eu[i]] = i;BITI b(n);ll res = 0;for(int i = n - 1; i >= 0; i--){res += b.sum(tr[i]+cs[i]);res -= b.sum(tr[i]);b.add(tr[i], 1);}cout << res << endl;return 0;}