結果

問題 No.778 クリスマスツリー
ユーザー Mister
提出日時 2020-08-02 16:57:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 191 ms / 2,000 ms
コード長 2,512 bytes
コンパイル時間 1,195 ms
コンパイル使用メモリ 91,940 KB
最終ジャッジ日時 2025-01-12 13:27:53
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#include <functional>
template <class T>
struct SegmentTree {
using Merger = std::function<T(T, T)>;
int length;
std::vector<T> dat;
T unit;
Merger merge;
SegmentTree() = default;
SegmentTree(int n, const T& unit, const Merger& merge)
: length(1), unit(unit), merge(merge) {
while (length < n) length <<= 1;
dat.assign(length * 2, unit);
}
template <class Container>
SegmentTree(const Container& elems, const T& unit, const Merger& merge)
: length(1), unit(unit), merge(merge) {
int n = elems.size();
while (length < n) length <<= 1;
dat.assign(length * 2, unit);
std::copy(elems.begin(), elems.end(), dat.begin() + length);
for (int nidx = length - 1; nidx >= 1; --nidx) {
T vl = dat[nidx * 2 + 0];
T vr = dat[nidx * 2 + 1];
dat[nidx] = merge(vl, vr);
}
}
void update(int nidx, const T& elem) {
nidx += length;
dat[nidx] = elem;
while (nidx > 0) {
nidx >>= 1;
T vl = dat[nidx * 2 + 0];
T vr = dat[nidx * 2 + 1];
dat[nidx] = merge(vl, vr);
}
}
T fold(int ql, int qr) const {
ql = std::max(ql, 0);
qr = std::min(qr, length);
ql += length, qr += length;
T lacc = unit, racc = unit;
while (ql < qr) {
if (ql & 1) {
lacc = merge(lacc, dat[ql]);
++ql;
}
if (qr & 1) {
--qr;
racc = merge(dat[qr], racc);
}
ql >>= 1, qr >>= 1;
}
return merge(lacc, racc);
}
T get(int idx) const { return dat[idx + length]; }
T fold_all() const { return dat[1]; }
};
using lint = long long;
void solve() {
int n;
std::cin >> n;
std::vector<std::vector<int>> graph(n);
for (int v = 1; v < n; ++v) {
int p;
std::cin >> p;
graph[p].push_back(v);
}
SegmentTree<lint>
seg(n, 0, [](auto a, auto b) { return a + b; });
lint ans = 0;
std::function<void(int)> dfs =
[&](int v) -> void {
ans += seg.fold(0, v);
seg.update(v, 1);
for (auto u : graph[v]) dfs(u);
seg.update(v, 0);
};
dfs(0);
std::cout << ans << "\n";
}
int main() {
std::cin.tie(nullptr);
std::ios::sync_with_stdio(false);
solve();
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0