結果

問題 No.778 クリスマスツリー
コンテスト
ユーザー face4
提出日時 2019-03-02 19:37:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 121 ms / 2,000 ms
コード長 851 bytes
コンパイル時間 797 ms
コンパイル使用メモリ 71,432 KB
実行使用メモリ 21,120 KB
最終ジャッジ日時 2024-10-14 01:58:33
合計ジャッジ時間 2,270 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
using namespace std;
typedef long long ll;

struct BIT{
    vector<int> node;
    int n;
    BIT(int siz){
        n = siz+1;
        node.resize(n, 0); 
    }

    void add(int i, int x){
        while(i < n){
            node[i] += x;
            i += i&-i;
        }
    }

    int sum(int i){
        int ret = 0;
        while(i > 0){
            ret += node[i];
            i -= i&-i;
        }
        return ret;
    }
};

BIT bit(200000);
vector<int> a[200000];
int n;
ll ans = 0;

void dfs(int cur){
    ans += bit.sum(cur+1);
    bit.add(cur+1, 1);
    for(int next : a[cur]){
        dfs(next);
    }
    bit.add(cur+1, -1);
}

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