結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー fumiphys
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

// 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;

//  types
typedef pair<int, int> P;
typedef pair<ll, int> Pl;
typedef pair<ll, ll> Pll;
typedef pair<double, double> Pd;
 
// constants
const int inf = 1e9;
const ll linf = 1LL << 50;
const double EPS = 1e-10;
const int mod = 1e9 + 7;

// solve
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(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;
}
0