結果

問題 No.778 クリスマスツリー
ユーザー first_vilfirst_vil
提出日時 2019-09-11 16:43:12
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 71 ms / 2,000 ms
コード長 2,236 bytes
コンパイル時間 2,077 ms
コンパイル使用メモリ 173,640 KB
実行使用メモリ 21,376 KB
最終ジャッジ日時 2024-10-14 02:00:34
合計ジャッジ時間 3,073 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 1 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 53 ms
21,376 KB
testcase_07 AC 26 ms
9,908 KB
testcase_08 AC 71 ms
15,144 KB
testcase_09 AC 69 ms
12,160 KB
testcase_10 AC 62 ms
12,024 KB
testcase_11 AC 67 ms
12,160 KB
testcase_12 AC 65 ms
12,032 KB
testcase_13 AC 41 ms
11,904 KB
testcase_14 AC 52 ms
21,368 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using VI = vector<int>;
using VL = vector<ll>;
#define FOR(i,a,n) for(int (i)=(a);(i)<(n);(i)++)
#define eFOR(i,a,n) for(int (i)=(a);(i)<=(n);(i)++)
#define rFOR(i,a,n) for(int (i)=(n)-1;(i)>=(a);(i)--)
#define erFOR(i,a,n) for(int (i)=(n);(i)>=(a);(i)--)
#define SORT(i) sort((i).begin(),(i).end())
#define rSORT(i,a) sort((i).begin(),(i).end(),(a))
#define all(i) (i).begin(),(i).end()
constexpr ll INF = 1000000000;
constexpr ll LLINF = 1LL << 60;
constexpr ll mod = 1000000007;
constexpr ll MOD = 998244353;
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; }return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; }return 0; }
inline void init() { cin.tie(nullptr); cout.tie(nullptr); cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); }

template<class V> class BIT {
    int n = 0; vector<V> bit;
#define def 0
public:
    BIT() {}  // [L, R) 半開区間・0-indexed
    BIT(int _n) { init(_n); }
    void init(int _n) { n = 1; while (n < _n)n *= 2; bit.resize(n, def); }
    V operator[](int e) { V s = 0; e++; while (e) s += bit[e - 1], e -= e & -e; return s; }
    void add(int e, V v) { e++; while (e <= n) bit[e - 1] += v, e += e & -e; }
    int lower_bound(V val) {
        V tv = 0; int i, ent = 0; for (i = n - 1; i >= 0; i--)
            if (tv + bit[ent + (1 << i) - 1] <= val)tv += bit[ent + (1 << i) - 1], ent += (1 << i); return ent;
    }
    V sum(int L, int R) {
        assert(0 <= L); assert(R <= n); assert(L <= R);
        V res = 0; if (R) res += operator[](R - 1); if (L) res -= operator[](L - 1); return res;
    }
    V sum(int R) { return sum(0, R); }
    void clear() { bit.clear(); }
    void resize(int _n) { bit.clear(); init(_n); }
    int size() { return n; }
};

BIT<int> bit;
vector<VI> edge;
ll ans;
void dfs(int i) {
    ans += bit.sum(i);
    bit.add(i, 1);
    for (int j : edge[i])dfs(j);
    bit.add(i, -1);
}

int main() {
    init();

    int n;
    cin >> n;
    edge.resize(n);
    bit.resize(n);
    int in;
    FOR(i, 1, n) {
        cin >> in;
        edge[in].push_back(i);
    }
    dfs(0);
    cout << ans << "\n";
}
0