結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー leaf_1415
提出日時 2019-01-11 03:47:50
言語 C++11
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=gnu++11 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 108 ms / 2,000 ms
コード長 806 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 645 ms
コンパイル使用メモリ 79,980 KB
実行使用メモリ 25,216 KB
最終ジャッジ日時 2026-05-01 12:30:55
合計ジャッジ時間 2,123 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#define llint long long

using namespace std;

struct BIT{
	int size;
	vector<llint> bit;
	BIT(){size = 0;}
	BIT(int s){
		size = s;
		bit.resize(size+1);
		init();
	}
	void init(){
		for(int i = 1; i <= size; i++) bit[i] = 0;
	}
	llint query(int i){
		int ret = 0;
		while(i > 0){
			ret += bit[i];
			i -= i&(-i);
		}
		return ret;
	}
	void add(int i, llint x){
		while(i <= size){
			bit[i] += x;
			i += i&(-i);
		}
	}
};

llint n;
vector<int> G[200005];
BIT bit(200005);

llint ans;
void dfs(int v)
{
	ans += bit.query(v);
	bit.add(v+1, 1);
	for(int i = 0; i < G[v].size(); i++) dfs(G[v][i]);
	bit.add(v+1, -1);
}

int main(void)
{
	cin >> n;
	int p;
	for(int i = 1; i < n; i++){
		cin >> p;
		G[p].push_back(i);
	}
	
	dfs(0);
	cout << ans << endl;
	
	return 0;
}
0