結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー chocorusk
提出日時 2018-12-25 00:18:08
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 112 ms / 2,000 ms
コード長 818 bytes
コンパイル時間 726 ms
コンパイル使用メモリ 97,828 KB
実行使用メモリ 24,448 KB
最終ジャッジ日時 2024-10-14 01:50:59
合計ジャッジ時間 2,245 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
int bit[200001], n;
int sum(int i){
	int s=0;
	while(i>0){
		s+=bit[i];
		i-=(i&(-i));
	}
	return s;
}
void add(int i, int x){
	while(i<=n){
		bit[i]+=x;
		i+=(i&(-i));
	}
}
vector<int> g[200001];
ll ans;
void dfs(int x){
	add(x, 1);
	ans+=(ll)sum(x-1);
	for(auto y:g[x]) dfs(y);
	add(x, -1);
}
int main()
{
	cin>>n;
	for(int i=1; i<n; i++){
		int a; cin>>a;
		g[a+1].push_back(i+1);
	}
	dfs(1);
	cout<<ans<<endl;
	return 0;
}
0