Problem p02647 (C++) — buggy submission, verdict: Time Limit Exceeded

/*
このコード、と～おれ!
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")
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep1(i, n) for (int i = 1; i <= (n); ++i)
#define rep2(i, n) for (int i = 2; i < (n); ++i)
#define repr(i, n) for (int i = n; i >= 0; --i)
#define reprm(i, n) for (int i = n - 1; i >= 0; --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 allsum(a, b, c) ((a + b) * c / 2LL) // 等差数列の和、初項,末項,項数
#define pb push_back
#define rpriq priq<int, vector<int>, greater<int>>
#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;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
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)); }

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

int main() {

  int n, k;
  scanf("%d%d", &n, &k);
  vector<int> a(n), b(n);
  rep(i, n) scanf("%d", &a[i]);
  if (k >= 75)
    rep(i, n) printf("%d\\n", n);
  else {
    while (k) {
      --k;
      fill(all(b), 0);
      rep(i, n) {
        int l = max(0, i - a[i]), r = min(n - 1, i + a[i]);
        ++b[l];
        if (r + 1 < n)
          --b[r + 1];
      }
      rep(i, n - 1) b[i + 1] += b[i];
      a = b;
    }
    rep(i, n) printf("%d\\n", a[i]);
  }

  Please AC;
}

Error: TLE

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")
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define rep1(i, n) for (int i = 1; i <= (n); ++i)
#define rep2(i, n) for (int i = 2; i < (n); ++i)
#define repr(i, n) for (int i = n; i >= 0; --i)
#define reprm(i, n) for (int i = n - 1; i >= 0; --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 allsum(a, b, c) ((a + b) * c / 2LL) // 等差数列の和、初項,末項,項数
#define pb push_back
#define rpriq priq<int, vector<int>, greater<int>>
#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;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
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)); }

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

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

// セグ木/0-indexed/非再帰/n の大きさ, a (単位元), 本体のマージ関数,
// 遅延ノードの単位元, 遅延ノードのマージ関数, 遅延ノードと本体のマージ関数 で
// segtree を初期化する
template <typename T, typename U> 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];
    }
  }

  void init() {
    fill(all(seg), se);
    fill(all(lazy), le);
    fill(all(flag), false);
  }

  // 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, T 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>
  // int find(const int start, const int end, int l, int r, int k, const C
  // check, T& checknum, const bool b, const function<T(T, T)> 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 >(start, end, l, ((l + r) >> 1), (k << 1) + 1, check,
  //checknum, b, revf);
  //		//左側が適してたらそれが答え
  //		if (res != -1)return (res);
  //		return find< C >(start, end, ((l + r) >> 1), r, (k << 1) + 2, check,
  //checknum, b, revf);
  //	}
  //	else {
  //		//右側を調べる
  //		res = find< C >(start, end, ((l + r) >> 1), r, (k << 1) + 2, check,
  //checknum, b, revf);
  //		//右側が適してたらそれが答え
  //		if (res != -1)return (res);
  //		return find< C >(start, end, l, ((l + r) >> 1), (k << 1) + 1, check,
  //checknum, b, revf);
  //	}
  // }

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

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

int main() {

  ll n, k;
  scanf("%lld%lld", &n, &k);
  vector<int> a(n);
  rep(i, n) scanf("%d", &a[i]);
  auto f = [](ll a, int w) { return a * (ll)w; };
  lazysegtree<ll, ll> seg(n, 0, plus<ll>(), 0, plus<ll>(), plus<ll>(), f);
  seg.build();
  if (k >= 75)
    rep(i, n) printf("%d\\n", n);
  else {
    while (k) {
      --k;
      seg.init();
      rep(i, n) {
        seg.update(max<ll>(0, i - a[i]), min<ll>(n, i + a[i] + 1), 1);
      }
      rep(i, n) { a[i] = seg[i]; }
    }
    rep(i, n) printf("%d\\n", a[i]);
  }

  Please AC;
}