結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー edamame882
提出日時 2019-06-24 09:38:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 69 ms / 2,000 ms
コード長 1,675 bytes
コンパイル時間 1,637 ms
コンパイル使用メモリ 173,996 KB
実行使用メモリ 28,212 KB
最終ジャッジ日時 2024-10-14 01:59:46
合計ジャッジ時間 2,784 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

//repetition
#define FOR(i,a,b) for(ll i=(a);i<(b);++i)
#define rep(i, n) for(ll i = 0; i < (ll)(n); i++)

//container util
#define all(x) (x).begin(),(x).end()

//typedef
typedef long long ll;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<ll> VLL;
typedef vector<VLL> VVLL;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef pair<ll, ll> PLL;

//const value
//const ll MOD = 1e9 + 7;
//const int dx[] = {0,1,0,-1};//{0,0,1,1,1,-1,-1,-1};
//const int dy[] = {1,0,-1,0};//{1,-1,0,1,-1,0,1,-1};

//conversion
inline int toInt(string s) {int v; istringstream sin(s);sin>>v;return v;}
inline ll toLL(string s) {ll v; istringstream sin(s);sin>>v;return v;}
template<class T> inline string toString(T x) {ostringstream sout;sout<<x;return sout.str();}

struct BIT{
  int n;
  VI nodes;
  void init(int N){
    nodes.resize(N+1,0);
    n = N;
  }
  void add(int i, int x){
    for(int idx = i; idx <= n; idx += idx & (-idx)){
      nodes[idx] += x;
    }
  }

  ll sum(int x){
    ll res = 0;
    for(int idx = x; idx > 0 ; idx -= idx & (-idx)){
      res += nodes[idx];
    }
    return res;
  }

};

ll dfs(int current_node, VVI& edge, BIT & bit){
  ll res = 0;
  res += bit.sum(current_node+1);
  bit.add(current_node+1,1);
  for(int node :edge[current_node]){
    res += dfs(node,edge,bit);
  }
  bit.add(current_node+1,-1);

  return res;
}
int main(){
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n;
  cin >> n;
  VI a(n);
  VVI edge(n);
  FOR(i,1,n){
    cin >> a[i];
    edge[a[i]].push_back(i);
  }

  BIT bit;
  bit.init(n);
  ll ans = dfs(0,edge,bit);
  cout << ans << endl;
  return 0;
}
0