結果
| 問題 |
No.2491 Pochi and A Warp Machine
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2025-05-17 19:36:47 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 22,628 bytes |
| コンパイル時間 | 6,214 ms |
| コンパイル使用メモリ | 332,040 KB |
| 実行使用メモリ | 338,312 KB |
| 最終ジャッジ日時 | 2025-05-17 19:38:08 |
| 合計ジャッジ時間 | 70,396 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 38 TLE * 2 |
ソースコード
// QCFium 法
//#pragma GCC target("avx2") // yukicoder では消す
#pragma GCC optimize("O3") // たまにバグる
#pragma GCC optimize("unroll-loops")
#ifndef HIDDEN_IN_VS // 折りたたみ用
// 警告の抑制
#define _CRT_SECURE_NO_WARNINGS
// ライブラリの読み込み
#include <bits/stdc++.h>
using namespace std;
// 型名の短縮
using ll = long long; using ull = unsigned long long; // -2^63 ~ 2^63 = 9e18(int は -2^31 ~ 2^31 = 2e9)
using pii = pair<int, int>; using pll = pair<ll, ll>; using pil = pair<int, ll>; using pli = pair<ll, int>;
using vi = vector<int>; using vvi = vector<vi>; using vvvi = vector<vvi>; using vvvvi = vector<vvvi>;
using vl = vector<ll>; using vvl = vector<vl>; using vvvl = vector<vvl>; using vvvvl = vector<vvvl>;
using vb = vector<bool>; using vvb = vector<vb>; using vvvb = vector<vvb>;
using vc = vector<char>; using vvc = vector<vc>; using vvvc = vector<vvc>;
using vd = vector<double>; using vvd = vector<vd>; using vvvd = vector<vvd>;
template <class T> using priority_queue_rev = priority_queue<T, vector<T>, greater<T>>;
using Graph = vvi;
// 定数の定義
const double PI = acos(-1);
int DX[4] = { 1, 0, -1, 0 }; // 4 近傍(下,右,上,左)
int DY[4] = { 0, 1, 0, -1 };
int INF = 1001001001; ll INFL = 4004004003094073385LL; // (int)INFL = INF, (int)(-INFL) = -INF;
// 入出力高速化
struct fast_io { fast_io() { cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(18); } } fastIOtmp;
// 汎用マクロの定義
#define all(a) (a).begin(), (a).end()
#define sz(x) ((int)(x).size())
#define lbpos(a, x) (int)distance((a).begin(), std::lower_bound(all(a), (x)))
#define ubpos(a, x) (int)distance((a).begin(), std::upper_bound(all(a), (x)))
#define Yes(b) {cout << ((b) ? "Yes\n" : "No\n");}
#define rep(i, n) for(int i = 0, i##_len = int(n); i < i##_len; ++i) // 0 から n-1 まで昇順
#define repi(i, s, t) for(int i = int(s), i##_end = int(t); i <= i##_end; ++i) // s から t まで昇順
#define repir(i, s, t) for(int i = int(s), i##_end = int(t); i >= i##_end; --i) // s から t まで降順
#define repe(v, a) for(const auto& v : (a)) // a の全要素(変更不可能)
#define repea(v, a) for(auto& v : (a)) // a の全要素(変更可能)
#define repb(set, d) for(int set = 0, set##_ub = 1 << int(d); set < set##_ub; ++set) // d ビット全探索(昇順)
#define repis(i, set) for(int i = lsb(set), bset##i = set; i < 32; bset##i -= 1 << i, i = lsb(bset##i)) // set の全要素(昇順)
#define repp(a) sort(all(a)); for(bool a##_perm = true; a##_perm; a##_perm = next_permutation(all(a))) // a の順列全て(昇順)
#define uniq(a) {sort(all(a)); (a).erase(unique(all(a)), (a).end());} // 重複除去
#define EXIT(a) {cout << (a) << endl; exit(0);} // 強制終了
#define inQ(x, y, u, l, d, r) ((u) <= (x) && (l) <= (y) && (x) < (d) && (y) < (r)) // 半開矩形内判定
// 汎用関数の定義
template <class T> inline ll powi(T n, int k) { ll v = 1; rep(i, k) v *= n; return v; }
template <class T> inline bool chmax(T& M, const T& x) { if (M < x) { M = x; return true; } return false; } // 最大値を更新(更新されたら true を返す)
template <class T> inline bool chmin(T& m, const T& x) { if (m > x) { m = x; return true; } return false; } // 最小値を更新(更新されたら true を返す)
template <class T> inline T getb(T set, int i) { return (set >> i) & T(1); }
template <class T> inline T smod(T n, T m) { n %= m; if (n < 0) n += m; return n; } // 非負mod
// 演算子オーバーロード
template <class T, class U> inline istream& operator>>(istream& is, pair<T, U>& p) { is >> p.first >> p.second; return is; }
template <class T> inline istream& operator>>(istream& is, vector<T>& v) { repea(x, v) is >> x; return is; }
template <class T> inline vector<T>& operator--(vector<T>& v) { repea(x, v) --x; return v; }
template <class T> inline vector<T>& operator++(vector<T>& v) { repea(x, v) ++x; return v; }
#endif // 折りたたみ用
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
#ifdef _MSC_VER
#include "localACL.hpp"
#endif
using mint = modint998244353;
//using mint = static_modint<(int)1e9 + 7>;
//using mint = modint; // mint::set_mod(m);
namespace atcoder {
inline istream& operator>>(istream& is, mint& x) { ll x_; is >> x_; x = x_; return is; }
inline ostream& operator<<(ostream& os, const mint& x) { os << x.val(); return os; }
}
using vm = vector<mint>; using vvm = vector<vm>; using vvvm = vector<vvm>; using vvvvm = vector<vvvm>; using pim = pair<int, mint>;
#endif
#ifdef _MSC_VER // 手元環境(Visual Studio)
#include "local.hpp"
#else // 提出用(gcc)
inline int popcount(int n) { return __builtin_popcount(n); }
inline int popcount(ll n) { return __builtin_popcountll(n); }
inline int lsb(int n) { return n != 0 ? __builtin_ctz(n) : 32; }
inline int lsb(ll n) { return n != 0 ? __builtin_ctzll(n) : 64; }
inline int msb(int n) { return n != 0 ? (31 - __builtin_clz(n)) : -1; }
inline int msb(ll n) { return n != 0 ? (63 - __builtin_clzll(n)) : -1; }
#define dump(...)
#define dumpel(v)
#define dump_math(v)
#define input_from_file(f)
#define output_to_file(f)
#define Assert(b) { if (!(b)) { vc MLE(1<<30); EXIT(MLE.back()); } } // RE の代わりに MLE を出す
#endif
//【グラフの入力】O(n + m)
/*
* (始点, 終点) の組からなる入力を受け取り,n 頂点 m 辺のグラフを構築して返す.
*
* n : グラフの頂点の数
* m : グラフの辺の数(省略すれば n-1)
* directed : 有向グラフか(省略すれば false)
* zero_indexed : 入力が 0-indexed か(省略すれば false)
*/
Graph read_Graph(int n, int m = -1, bool directed = false, bool zero_indexed = false) {
// verify : https://atcoder.jp/contests/tessoku-book/tasks/tessoku_book_bi
Graph g(n);
if (m == -1) m = n - 1;
rep(j, m) {
int a, b;
cin >> a >> b;
if (!zero_indexed) { --a; --b; }
g[a].push_back(b);
if (!directed && a != b) g[b].push_back(a);
}
return g;
}
//【木の重心分解】
/*
* Centroid_decomposition<G>(G g) : O(n log n)
* 木 g に対して,各部分木から重心を取り除く操作を繰り返して得られる根付き木を構築する.
* s の子が {t} ⇔ s を取り除いてできた {部分木} の重心が {t}
*
* int lca(int v1, int v2) : O(log n)
* 根付き木における頂点 v1, v2 の最小共通祖先 LCA(v1, v2) を返す.
* 性質:LCA(v1, v2) は元の木の v1-v2 パス上に存在する.
*/
template <class G>
struct Centroid_decomposition {
// 参考 : https://ferin-tech.hatenablog.com/entry/2020/03/06/162311
struct Node {
int size = -1; // この頂点を重心とする部分木の大きさ
int dep = -1; // この頂点が何回目の操作で取り除かれたか
int p = -1; // 親(なければ -1)
vi cs; // 子のリスト
int id = -1; // 親にとって何番目の子か
#ifdef _MSC_VER
friend ostream& operator<<(ostream& os, const Node& v) {
os << "size:" << v.size << ", ";
os << "dep:" << v.dep << ", ";
os << "p:" << v.p << ", ";
os << "cs:[" << v.cs << "], ";
os << "id:" << v.id << "";
return os;
}
#endif
};
int n; // 頂点の数
int rt; // 根
vector<Node> v; // 頂点
// 木 g で初期化する.
Centroid_decomposition(const G& g) : n(sz(g)), rt(-1), v(n) {
// verify : https://judge.yosupo.jp/problem/vertex_add_range_contour_sum_on_tree
// cnt[v] : 部分木 v の大きさ(使いまわす.根はその都度直前に取り除かれた重心に変わる)
vi cnt(n);
// bc を根としたときの部分木 s の大きさ cnt[s] を更新し,既に重心を発見しているかを返す.
// p : bc を根としたときの s の親
// r_size : s を含む部分木全体の大きさ
// dep : 何回目の処理か
// bc : 直前に取り除かれた重心
function<bool(int, int, int, int, int)> dfs = [&](int s, int p, int r_size, int dep, int bc) {
// 部分木 s の大きさ
cnt[s] = 1;
// ok : 頂点 s が重心か
bool ok = true;
// s の子 t を調べる.
repe(t, g[s]) {
// 親や既に取り除かれた頂点には進まない.
if (t == p || v[t].dep != -1) continue;
// 部分木 t の大きさを求める(既に重心を発見していればすぐに帰る)
if (dfs(t, s, r_size, dep, bc)) return true;
// 大きさが元の半分を超える部分木があれば s は重心ではない.
if (cnt[t] > r_size / 2) ok = false;
// 部分木 t の大きさを加える.
cnt[s] += cnt[t];
}
// p を含む部分木の大きさが元の半分を超えていれば s は重心ではない.
if (r_size - cnt[s] > r_size / 2) ok = false;
// s は重心なのでそれを記録し取り除く
if (ok) {
// s の情報を決定する.
v[s].size = r_size;
v[s].dep = dep;
v[s].p = bc;
if (bc != -1) {
v[s].id = sz(v[bc].cs);
v[bc].cs.push_back(s);
}
else rt = s;
// s の親 p を含む部分木を重心分解する.
if (p != -1) dfs(p, -1, r_size - cnt[s], dep + 1, s);
// s の各子 t を含む部分木を重心分解する.
repe(t, g[s]) {
if (t == p || v[t].dep != -1) continue;
dfs(t, -1, cnt[t], dep + 1, s);
}
return true;
}
return false;
};
dfs(0, -1, sz(g), 0, -1);
}
Centroid_decomposition() : n(0), rt(-1) {}
// アクセス
Node const& operator[](int i) const { return v[i]; }
Node& operator[](int i) { return v[i]; }
// 大きさ
int size() const { return n; }
// v1 と v2 の最小共通祖先を返す.
int lca(int v1, int v2) {
// 木の高さが O(log n) なので,ダブリングを用いず愚直に上っていっても高速
while (v[v1].dep < v[v2].dep) v2 = v[v2].p;
while (v[v1].dep > v[v2].dep) v1 = v[v1].p;
while (v1 != v2) { v1 = v[v1].p; v2 = v[v2].p; }
return v1;
}
#ifdef _MSC_VER
friend ostream& operator<<(ostream& os, const Centroid_decomposition& cd) {
rep(i, sz(cd)) os << i << ": " << cd[i] << endl;
return os;
}
#endif
};
//【オイラーツアー】
/*
* Euler_tour(Graph g, int rt) : O(n)
* rt を根とする根付き木 g で初期化する.
*
* int lca(int s, int t) : O(log n)
* 頂点 s, t の最小共通祖先を返す.
*
* int dist(int s, int t) : O(log n)
* 頂点 s, t 間の距離を返す.
*
* int jump(int s, int t, int i) : O(log n)
* 頂点 s から t までのパスの i 番目(0-indexed)の頂点を返す(なければ -1)
*
* sort_by_DFS_order(vi& vs) : O(log |vs|)
* 頂点集合 vs を DFS 昇順にソートする.
*/
ll op_ET(ll a, ll b) { return min(a, b); }
ll e_ET() { return INFL; }
struct Euler_tour {
int n;
// in[s] : rt からの DFS で最初に頂点 s を訪れた時刻(根なら 0)
// out[s] : rt からの DFS で最後に頂点 s から離れた時刻(根なら 2n-1)
// pos[t] : rt からの DFS で時刻 t に居た頂点の番号(長さ 2n-1)
// dep[s] : 頂点 s の深さ
vi in, out, pos, dep;
// seg[t] : 時刻 t に居た頂点の (深さ, 番号)
using SEG = segtree<ll, op_ET, e_ET>;
SEG seg;
void dfs(const Graph& g, int rt) {
int time = 0;
function<void(int, int)> rf = [&](int s, int p) {
// s を最初に訪れた
in[s] = time;
pos[time] = s;
time++;
repe(t, g[s]) {
if (t == p) continue;
dep[t] = dep[s] + 1;
rf(t, s);
pos[time] = s;
time++;
}
// s から最後に離れる
out[s] = time;
};
// 根から順に探索する.
rf(rt, -1);
}
public:
// rt を根とする根付き木 g で初期化する.
Euler_tour(const Graph& g, int rt) : n(sz(g)), in(n), out(n), pos(2 * n - 1), dep(n) {
// verify : https://judge.yosupo.jp/problem/lca
dfs(g, rt);
vl ini(2 * n - 1);
rep(t, 2 * n - 1) ini[t] = ((ll)dep[pos[t]] << 32) | pos[t];
seg = SEG(ini);
}
Euler_tour() : n(0) {}
// 頂点 s, t の最小共通祖先を返す.
int lca(int s, int t) const {
// verify : https://judge.yosupo.jp/problem/lca
// 初めて s または t に訪れたとき
int l = min(in[s], in[t]);
// 最後に s または t から離れたとき
int r = max(out[s], out[t]);
// その途中で訪れたことのある最も浅い頂点が最小共通祖先
return (int)seg.prod(l, r);
}
// 頂点 s, t 間の距離を返す.
int dist(int s, int t) const {
// verify : https://yukicoder.me/problems/no/2337
int p = lca(s, t);
// 根からの距離(深さ)の和を求め,ダブっている分を引く.
return dep[s] + dep[t] - 2 * dep[p];
}
// 頂点 s から t までのパスの i 番目(0-indexed)の頂点を返す(なければ -1)
int jump(int s, int t, int i) const {
// verify : https://judge.yosupo.jp/problem/jump_on_tree
int p = lca(s, t);
int ds = dep[s], dt = dep[t], dp = dep[p];
int dist = ds + dt - 2 * dp;
int res;
if (i < 0 || i > dist) {
res = -1;
}
else if (i <= ds - dp) {
int j = seg.max_right(out[s] - 1, [&](ll tmp) { return (tmp >> 32) > ds - i; });
res = pos[j];
}
else {
int j = seg.min_left(in[t] + 1, [&](ll tmp) { return (tmp >> 32) >= dt - (dist - i); });
res = pos[j];
}
return res;
}
// 頂点集合 vs を DFS 昇順にソートする.
void sort_by_DFS_order(vi& vs) const {
// verify : https://yukicoder.me/problems/no/2588
sort(all(vs), [&](int s, int t) { return in[s] < in[t]; });
}
};
//【フェニック木(アーベル群)】(の改変)
/*
* Fenwick_tree<S, op, o, inv>(int n) : O(n)
* v[0..n) = o() で初期化する.要素はアーベル群 (S, op, o, inv) の元とする.
*
* Fenwick_tree<S, op, o, inv>(vS a) : O(n)
* v[0..n) = a[0..n) で初期化する.
*
* set(int i, S x) : O(log n)
* v[i] = x とする.
*
* S get(int i) : O(log n)
* v[i] を返す.
*
* S sum(int l, int r) : O(log n)
* Σv[l..r) を返す.空なら o() を返す.
*
* add(int i, S x) : O(log n)
* v[i] += x とする.
*
* int max_right(function<bool(S)>& f) : O(log n)
* f( Σv[0..r) ) = true となる最大の r を返す.
* 制約:f( o() ) = true,f は単調
*/
template <class S, S(*op)(S, S), S(*o)(), S(*inv)(S)>
class Fenwick_tree {
// 参考:https://algo-logic.info/binary-indexed-tree/
// n : 要素数
int n;
// v[i] : Σa[*..i] の値(i:1-indexed,v[0] は不使用)
vector<S> v;
public:
// Σa[1..r] を返す.空なら o() を返す.(r:1-indexed)
S sum_sub(int r) const {
S res = o();
// 根に向かって累積 op() をとっていく.
while (r > 0) {
res = op(res, v[r]);
// r の最下位ビットを 0 にすることで次の位置を得る.
r -= r & -r;
}
return res;
}
// a[0..n) = o() で初期化する.
Fenwick_tree(int n) : n(n), v(n + 1, o()) {
// verify : https://judge.yosupo.jp/problem/range_kth_smallest
}
// 配列 a[0..n) で初期化する.
Fenwick_tree(const vector<S>& a) : n(sz(a)), v(n + 1) {
// verify : https://judge.yosupo.jp/problem/point_add_range_sum
// 配列の値を仮登録する.
rep(i, n) v[i + 1] = a[i];
// 正しい値になるよう根に向かって累積 op() をとっていく.
for (int pow2 = 1; 2 * pow2 <= n; pow2 *= 2) {
for (int i = 2 * pow2; i <= n; i += 2 * pow2) {
v[i] = op(v[i], v[i - pow2]);
}
}
}
Fenwick_tree() : n(0) {}
// a[i] = x とする.(i : 0-indexed)
void set(int i, S x) {
Assert(0 <= i && i < n);
// 差分を求める.
S d = op(x, inv(get(i)));
add(i, d);
}
// a[i] を返す.(i : 0-indexed)
S get(int i) const {
Assert(0 <= i && i < n);
return sum(i, i + 1);
}
// Σa[l..r) を返す.空なら o() を返す.(l, r : 0-indexed)
S sum(int l, int r) const {
// verify : https://judge.yosupo.jp/problem/point_add_range_sum
chmax(l, 0); chmin(r, n);
if (l >= r) return o();
// 0-indexed での半開区間 [l, r) は,
// 1-indexed での閉区間 [l + 1, r] に対応する.
// よって閉区間 [1, r] の総和から閉区間 [1, l] の総和を引けば良い.
return op(sum_sub(r), inv(sum_sub(l)));
}
// a[i] += x とする.(i : 0-indexed)
void add(int i, S x) {
// verify : https://judge.yosupo.jp/problem/point_add_range_sum
Assert(0 <= i && i < n);
// i を 1-indexed に直す.
i++;
// 根に向かって値を op() していく.
while (i <= n) {
v[i] = op(v[i], x);
// i の最下位ビットに 1 を加算することで次の位置を得る.
i += i & -i;
}
}
// f( Σa[0..r) ) = true となる最大の r を返す.(r : 0-indexed)
int max_right(const function<bool(S)>& f) const {
// verify : https://www.spoj.com/problems/ALLIN1/
S x = o();
// 注目している閉区間は [l+1, r] で幅は len
int l = 0;
for (int len = 1 << msb(n); len > 0; len >>= 1) {
int r = l + len;
if (r > n) continue;
auto nx = op(x, v[r]);
if (f(nx)) {
x = nx;
l = r;
}
}
return l;
}
#ifdef _MSC_VER
friend ostream& operator<<(ostream& os, const Fenwick_tree& ft) {
rep(i, ft.n) {
os << ft.get(i) << " ";
}
return os;
}
#endif
};
//【組の総和 アーベル群】
using S612 = pair<ll, int>;
S612 op612(S612 a, S612 b) { return { a.first + b.first, a.second + b.second }; }
S612 e612() { return { 0, 0 }; }
S612 inv612(S612 a) { return { -a.first, -a.second }; }
#define PairSum_group S612, op612, e612, inv612
//【矩形加算 → 1 点参照(一括)】(の改変)
/*
* Offline_rectangle_add<T>() : O(1)
* v[0..h)[0..w) = 0 で初期化する(h, w は自動で調整される)
*
* void rectangle_add(ll x1, ll x2, ll y1, ll y2, T val) : O(1)
* v[x1..x2)[y1..y2) += val とする.
*
* void get(ll x, ll y) : O(1)
* クエリ v[x][y] を追加する.
*
* vT solve() : O((n + q) log n)
* 現在の v[0..h)[0..w) への各クエリに対する答えを格納したリストを返す.
*
*(平面走査)
*/
template <class T>
class Offline_rectangle_add {
vl x2_add, y2_add; vector<T> w_add;
vl x_get, y_get;
public:
vector<T> res; // 中で持ったほうが扱いやすい?
int pt = 0;
// v[0..h)[0..w) = 0 で初期化する(h, w は自動で調整される)
Offline_rectangle_add() {
// verify : https://mojacoder.app/users/Tonegawac/problems/plane_add
}
// v[x1..x2)[y1..y2) += val とする.
void rectangle_add(ll x2, ll y2, T val) {
// verify : https://mojacoder.app/users/Tonegawac/problems/plane_add
//x1_add.emplace_back(x1);
x2_add.emplace_back(x2);
//y1_add.emplace_back(y1);
y2_add.emplace_back(y2);
w_add.emplace_back(val);
}
// クエリ v[x][y] を追加する.
void get(ll x, ll y) {
// verify : https://mojacoder.app/users/Tonegawac/problems/plane_add
x_get.emplace_back(x);
y_get.emplace_back(y);
}
// 各クエリに対する答えを格納したリストを返す.
void solve() {
// verify : https://mojacoder.app/users/Tonegawac/problems/plane_add
// ys : y 座標のユニークな昇順列
vl ys;
ys.reserve(1 + sz(y2_add));
//repe(y, y1_add) ys.emplace_back(y);
ys.emplace_back(0);
repe(y, y2_add) ys.emplace_back(y);
uniq(ys);
Fenwick_tree<PairSum_group> fen(sz(ys));
// (x 座標, クエリ番号, y 座標, 加算時の符号) の組(加算は負のクエリ番号で表す)
vector<tuple<ll, int, int, int>> ev;
rep(i, sz(x2_add)) {
//ev.emplace_back(0, -i - 1, 0, 1);
//ev.emplace_back(0, -i - 1, lbpos(ys, y2_add[i]), -1);
fen.add(lbpos(ys, y2_add[i]), { -w_add[i], -1 });
ev.emplace_back(x2_add[i], -i - 1, 0, -1);
ev.emplace_back(x2_add[i], -i - 1, lbpos(ys, y2_add[i]), 1);
}
int q = sz(x_get);
rep(t, q) {
ev.emplace_back(x_get[t], t, ubpos(ys, y_get[t]), 1);
}
S612 fen0 = { accumulate(all(w_add), 0LL), sz(w_add) };
// イベントソート
sort(all(ev));
res.resize(q);
// 下方向に平面走査していく.
for (auto& [x, id, y, sgn] : ev) {
if (id < 0) {
// 加算する場合
if (y == 0) {
fen0.first += sgn * w_add[-id - 1];
fen0.second += sgn;
}
else {
fen.add(y, { sgn * w_add[-id - 1], sgn });
}
}
else {
// 取得クエリの場合
auto [s0, s1] = fen.sum_sub(y);
s0 += fen0.first;
s1 += fen0.second;
res[id] = s0 + s1 * x;
}
}
}
};
int main() {
// input_from_file("input.txt");
// output_to_file("output.txt");
// https://judge.yosupo.jp/problem/vertex_get_range_contour_add_on_tree
int n;
cin >> n;
auto g = read_Graph(n);
Centroid_decomposition<Graph> CD(g);
//dump(CD);
Euler_tour ET(g, 0);
vi dist(n - 1);
rep(i, n - 1) dist[i] = ET.dist(i, i + 1);
//dump("dist:", dist);
//using F = Fenwick_tree_range_add<Sum_Zmodule>;
//vector<F> fen(n);
//rep(s, n) fen[s] = F(CD[s].size + 1); // 実際は半分で良い
//vector<vector<F>> fen_sub(n);
//rep(s, n) {
// int K = sz(CD[s].cs);
// fen_sub[s].resize(K);
// rep(k, K) {
// fen_sub[s][k] = F(CD[CD[s].cs[k]].size + 1);
// }
//}
using T = ll;
using F = Offline_rectangle_add<T>;
vector<F> fen(n);
vector<vector<F>> fen_sub(n);
rep(s, n) {
int K = sz(CD[s].cs);
fen_sub[s].resize(K);
}
vector<queue<int>> d0s(n);
auto add = [&](int s, int d_ub, int id_ub, ll val) {
int p = s; int id = -1;
while (p != -1) {
int d0 = ET.dist(s, p);
d0s[s].push(d0);
int d = d_ub - d0;
//fen[p].add(0, d, val);
fen[p].rectangle_add(d, id_ub, val + d0);
//dump("add:", s, p, d, id_ub, val + d0);
if (id != -1) {
//fen_sub[p][id].add(0, d, val);
fen_sub[p][id].rectangle_add(d, id_ub, val + d0);
}
id = CD[p].id;
p = CD[p].p;
}
};
repi(i, 1, n - 1) {
add(i, dist[i - 1] - 1, i, 1 - dist[i - 1]);
}
auto get = [&](int s) {
int p = s; int id = -1; ll res = 0;
while (p != -1) {
int d;
if (s == 0) {
d = ET.dist(s, p);
}
else {
d = d0s[s].front(); d0s[s].pop();
}
fen[p].get(d, s);
if (id != -1) {
fen_sub[p][id].get(d, s);
}
id = CD[p].id;
p = CD[p].p;
}
};
rep(i, n) get(i);
rep(i, n) {
fen[i].solve();
int K = sz(fen_sub[i]);
rep(k, K) fen_sub[i][k].solve();
}
auto get2 = [&](int s) {
int p = s; int id = -1; ll res = 0;
while (p != -1) {
T tmp = fen[p].res[fen[p].pt++];
res += tmp;
if (id != -1) {
tmp = fen_sub[p][id].res[fen_sub[p][id].pt++];
res -= tmp;
}
id = CD[p].id;
p = CD[p].p;
}
return res;
};
ll sc0 = accumulate(all(dist), 0LL);
rep(i, n) cout << sc0 + get2(i) << "\n";
}