Problem p02568 (C++) — buggy submission, verdict: Runtime Error

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;

#pragma gcc target("avx2")
#pragma gcc optimization("o3")
#pragma gcc optimization("unroll-loops")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
 * @title lazy-segment-tree
 * @docs kyopro/docs/lazysegtree.md
 */

// セグ木/0-indexed/非再帰/n の大きさ, a (単位元), 本体のマージ関数,
// 遅延ノードの単位元, 遅延ノードのマージ関数, 遅延ノードと本体のマージ関数 で
// segtree を初期化する
template <typename T, typename U, typename F, typename F2, typename F3,
          typename F4>
struct lazysegtree {
  // 木を配列であらわしたもの
  vector<T> seg;
  // 遅延ノード
  vector<U> lazy;
  // サイズノード
  vector<int> size;
  // 遅延ノードのフラグ管理
  vector<bool> flag;
  // 木の1/2の大きさ
  int siz, height;
  // 本体の単位元
  const T se;
  // 遅延ノードの単位元
  const U le;
  ////本体のマージ関数の型
  // using F = function<T(T, T)>;
  ////遅延ノードのマージ関数の型
  // using F2 = function<U(U, U)>;
  ////遅延ノードと本体のマージ関数の型
  // using F3 = function<T(T, U)>;
  ////サイズを使った演算をする関数の型
  // using F4 = function<U(U, int)>;
  // 本体同士をマージする関数
  const F f;
  // 遅延ノード同士をマージする関数
  const F2 f2;
  // 遅延ノードと本体をマージする関数
  const F3 f3;
  // サイズを使った演算をする関数
  const F4 f4;

  // n の大きさ, a (単位元), 本体のマージ関数, 遅延ノードの単位元,
  // 遅延ノードのマージ関数, 遅延ノードと本体のマージ関数,
  // サイズを使った演算をする関数 で segtree を初期化する
  lazysegtree(int n, const T se, const F f, const U le, const F2 f2,
              const F3 f3, const F4 f4)
      : se(se), f(f), le(le), f2(f2), f3(f3), f4(f4) {
    siz = 1;
    height = 0;
    while (siz < n)
      siz <<= 1, ++height;
    seg.assign(2 * siz - 1, se);
    lazy.assign(2 * siz - 1, le);
    size.assign(2 * siz - 1, 1);
    flag.assign(2 * siz - 1, false);
    --siz;
  }

  // k (0-indexed) 番目に t を代入
  void set(int k, const T &t) { seg[k + siz] = t; }

  // f によって木を構築
  void build() {
    for (int i = siz - 1; i >= 0; --i) {
      seg[i] = f(seg[i * 2 + 1], seg[i * 2 + 2]);
      size[i] = size[i * 2 + 1] + size[i * 2 + 2];
    }
  }

  // i 番目の要素を返す
  T operator[](const int i) { return query(i, i + 1); }

  // k 番目の遅延ノードを伝播する
  inline T merge(int k) {
    return (!flag[k] ? seg[k] : f3(seg[k], f4(lazy[k], size[k])));
  }

  // 子に伝播
  inline void eval(int k) {
    if (flag[k]) {
      lazy[k * 2 + 1] = f2(lazy[k * 2 + 1], lazy[k]);
      lazy[k * 2 + 2] = f2(lazy[k * 2 + 2], lazy[k]);
      flag[k * 2 + 1] = flag[k * 2 + 2] = true;
      seg[k] = merge(k);
      lazy[k] = le;
      flag[k] = false;
    }
  }

  inline void bottomup(int k) {
    while (k > 0) {
      k = ((k - 1) >> 1);
      seg[k] = f(merge(2 * k + 1), merge(2 * k + 2));
    }
  }

  inline void topdown(int k) {
    for (int i = height; i > 0; --i) {
      eval(((k + 1) >> i) - 1);
    }
  }

  // k 番目の値を a に更新
  void update(int k, T a) {
    k += siz;
    // 必要であればここを変える
    seg[k] = a;
    while (k > 0) {
      k = ((k - 1) >> 1);
      seg[k] = f(seg[k * 2 + 1], seg[k * 2 + 2]);
    }
  }

  //[l, r) の値を a に更新
  void update(int l, int r, U a) {
    int x = l + siz, y = r + siz - 1;
    topdown(x);
    topdown(y);
    for (l += siz, r += siz; l < r; l >>= 1, r >>= 1) {
      if (!(l & 1)) {
        lazy[l] = f2(lazy[l], a);
        flag[l] = true;
        ++l;
      }
      if (!(r & 1)) {
        --r;
        lazy[r] = f2(lazy[r], a);
        flag[r] = true;
      }
    }
    bottomup(x);
    bottomup(y);
  }

  //[a, b) について f した結果を返す
  T query(int a, int b) {
    topdown(a += siz);
    topdown(b += siz - 1);
    T l = se, r = se;
    for (++b; a < b; a >>= 1, b >>= 1) {
      if (!(a & 1))
        l = f(l, merge(a++));
      if (!(b & 1))
        r = f(merge(--b), r);
    }
    return f(l, r);
  }

  ////[start, end) について、[l, r) を調べながら k 番目が check
  ///を満たすか二分探索 最後が true なら left, false なら right fの逆演算
  // template<typename C, typename FT>
  // int find(const int start, const int end, int l, int r, int k, const C
  // check, T& checknum, const bool b, const FT revf) {
  //	//cerr << checknum << '\\n';
  //	//範囲外またはそこがすでに満たさないとき
  //	//cerr << k << ',' << checknum << '\\n';
  //	if (start <= l && r <= end && !check(seg[k], checknum)) {
  //		checknum = revf(checknum, seg[k]);
  //		return -1;
  //	}
  //	if ((r <= start || l >= end)) {
  //		return -1;
  //	}
  //	//既に葉
  //	if (k >= siz) {
  //		return k - siz;
  //	}
  //	int res;
  //	if (b) {
  //		//左側を調べる
  //		res = find< C, FT >(start, end, l, ((l + r) >> 1), (k << 1) + 1,
  //check, checknum, b, revf);
  //		//左側が適してたらそれが答え
  //		if (res != -1)return (res);
  //		return find< C, FT >(start, end, ((l + r) >> 1), r, (k << 1) + 2,
  //check, checknum, b, revf);
  //	}
  //	else {
  //		//右側を調べる
  //		res = find< C, FT >(start, end, ((l + r) >> 1), r, (k << 1) + 2,
  //check, checknum, b, revf);
  //		//右側が適してたらそれが答え
  //		if (res != -1)return (res);
  //		return find< C, FT >(start, end, l, ((l + r) >> 1), (k << 1) + 1,
  //check, checknum, b, revf);
  //	}
  // }

  // template<typename C, typename FT>
  // int find_left(int start, int end, const C check, T checknum, FT revf) {
  //	return find< C, FT >(start, end, 0, siz + 1, 0, check, checknum, true,
  //revf);
  // }

  // template<typename C, typename FT>
  // int find_right(int start, int end, const C check, T checknum, FT revf) {
  //	return find< C, FT >(start, end, 0, siz + 1, 0, check, checknum, false,
  //revf);
  // }
};

/*
 * @title modint
 * @docs kyopro/docs/modint.md
 */

template <int mod> struct modint {
  int val;

  modint() : val(0){};
  modint(ll x) : val(x >= 0 ? x % mod : (mod + x % mod) % mod){};

  modint &operator=(const modint &x) {
    val = x.val;
    return *this;
  }

  modint &operator+=(const modint &x) {
    val += x.val;
    if (val >= mod)
      val -= mod;
    return *this;
  }

  modint &operator-=(const modint &x) {
    val += mod - x.val;
    if (val >= mod)
      val -= mod;
    return *this;
  }

  modint &operator*=(const modint &x) {
    val = (int)((ll)val * (ll)x.val % mod);
    return *this;
  }

  modint &operator/=(const modint &x) {
    int a = x.val, b = mod, u = 1, v = 0, t;
    while (b > 0) {
      t = a / b;
      swap(a -= t * b, b);
      swap(u -= t * v, v);
    }
    *this *= modint(u);
    return *this;
  }

  modint operator++() {
    val = (val + 1 == mod ? 0 : val + 1);
    return *this;
  }

  modint operator--() {
    val = (val == 0 ? mod - 1 : val - 1);
    return *this;
  }

  modint operator+(const modint &x) const { return (modint(*this) += x); }

  modint operator-(const modint &x) const { return (modint(*this) -= x); }

  modint operator*(const modint &x) const { return (modint(*this) *= x); }

  modint operator/(const modint &x) const { return (modint(*this) /= x); }

  bool operator==(const modint &x) const { return (val == x.val); }

  bool operator!=(const modint &x) const { return (val != x.val); }

  bool operator<(const modint &x) const { return (val < x.val); }

  bool operator>(const modint &x) const { return (val > x.val); }

  modint pow(ll n) {
    modint ret(1), a(val);
    while (n > 0) {
      if (n % 2)
        ret *= a;
      a *= a;
      n /= 2;
    }
    return ret;
  }

  static int getmod() { return mod; };
};

using ModInt = modint<MOD>;
using Modint = modint<mod>;

inline auto f2 = [](pair<Modint, Modint> a, pair<Modint, Modint> b) {
  return make_pair(a.first * b.first, a.second * b.first + b.second);
};
inline auto f3 = [](Modint a, pair<Modint, Modint> b) {
  return a * b.first + b.second;
};
inline auto f4 = [](pair<Modint, Modint> a, int siz) {
  return make_pair(a.first, a.second * siz);
};

int main() {

  int n, q;
  scanf("%d%d", &n, &q);
  lazysegtree<Modint, pair<Modint, Modint>, decltype(plus<Modint>()),
              decltype(f2), decltype(f3), decltype(f4)>
      seg(n, 0, plus<Modint>(), make_pair(1, 0), f2, f3, f4);
  Modint tmp;
  rep(i, n) {
    scanf("%d", &tmp.val);
    seg.set(i, tmp);
  }
  seg.build();
  int type, l, r, b, c;
  while (q--) {
    scanf("%d%d%d", &type, &l, &r);
    if (type) {
      printf("%d\\n", seg.query(l, r).val);
    } else {
      scanf("%d%d", &b, &c);
      seg.update(l, r, make_pair(b, c));
    }
  }

  Please AC;
}


Error: -11

Fixed submission:

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;

// #pragma GCC target ("avx2")
// #pragma GCC optimization ("Ofast")
// #pragma GCC optimization ("unroll-loops")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> inline T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> inline T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
 * @title template(graph)
 * @docs kyopro/docs/graph_template.md
 */

template <typename T> struct edge {
  T cost;
  int from, to;

  edge(int from, int to) : from(from), to(to), cost(T(1)) {}
  edge(int from, int to, T cost) : from(from), to(to), cost(cost) {}
};

template <typename T = int> struct graph {

  int n;
  bool directed, weighted;

  vector<vector<edge<T>>> g;

  graph(int n, bool directed, bool weighted)
      : g(n), n(n), directed(directed), weighted(weighted) {}

  void add_edge(int from, int to, T cost = T(1)) {
    g[from].emplace_back(from, to, cost);
    if (not directed) {
      g[to].emplace_back(to, from, cost);
    }
  }

  vector<edge<T>> &operator[](const int &idx) { return g[idx]; }

  void read(int e, bool one_indexed) {
    int a, b, c = 1;
    while (e--) {
      scanf("%d%d", &a, &b);
      if (weighted) {
        scanf("%d", &c);
      }
      if (one_indexed)
        --a, --b;
      add_edge(a, b, c);
    }
  }

  void read(int e, bool one_indexed, const string &format) {
    int a, b;
    T c = T(1);
    while (e--) {
      scanf("%d%d", &a, &b);
      if (weighted) {
        scanf(format.c_str(), &c);
      }
      if (one_indexed)
        --a, --b;
      add_edge(a, b, c);
    }
  }
};

template <typename T = int>
void diameter1(graph<T> &g, const int &cur, const int &bef, const T &cnt, T &ma,
               int &idx) {
  if (cnt > ma)
    ma = cnt, idx = cur;
  for (const auto &aa : g[cur])
    if (aa.to != bef)
      diameter1<T>(g, aa.to, cur, cnt + aa.cost, ma, idx);
}

template <typename T = int>
void diameter2(graph<T> &g, const int &cur, const int &bef, vector<int> &ret,
               vector<int> &route, const T &cnt, T &cost) {
  route.emplace_back(cur);
  if (g[cur].size() == 1 and cost < cnt) {
    cost = cnt;
    ret = route;
  }
  for (const auto &aa : g[cur])
    if (aa.to != bef)
      diameter2<T>(g, aa.to, cur, ret, route, cnt + aa.cost, cost);
  route.pop_back();
}

template <typename T = int>
vector<int> diameter(graph<T> &g, T &cost, const int &cur = 0) {
  T ma = 0;
  int idx = cur;
  diameter1<T>(g, cur, -1, (T)0, ma, idx);
  ma = 0;
  vector<int> ret, route;
  diameter2<T>(g, idx, -1, ret, route, 0, cost);
  reverse(all(ret));
  return ret;
}

int main() {

  int n;
  scanf("%d", &n);
  graph<ll> g(n, false, true);
  g.read(n - 1, false, "%lld");
  ll ans = 0;
  auto v = diameter<ll>(g, ans);
  printf("%lld %d\\n", ans, (int)v.size());
  for (const auto &aa : v)
    printf("%d ", aa);

  Please AC;
}