結果

問題 No.778 クリスマスツリー
ユーザー りあん
提出日時 2018-12-25 00:44:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 607 bytes
コンパイル時間 2,049 ms
コンパイル使用メモリ 203,024 KB
最終ジャッジ日時 2025-01-06 20:00:29
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
const int M = 1000000007;

vector<vector<int>> edge;
set<int> se;

long long dfs(int p, int par) {
    long long ret = distance(se.begin(), se.lower_bound(p));
    se.insert(p);
    for (auto& i : edge[p]) {
        if (i == par) continue;
        ret += dfs(i, p);
    }
    se.erase(p);
    return ret;
}

int main() {
    int n;
    cin >> n;
    edge = vector<vector<int>>(n);
    for (int i = 1; i < n; ++i) {
        int a;
        cin >> a;
        edge[i].push_back(a);
        edge[a].push_back(i);
    }
    cout << dfs(0, -1) << "\n";
    return 0;
}
0