Problem p02954 (C++)

Buggy submission (status: Time Limit Exceeded):
/* #region Head */

//============================================================================
// Name        : filename.cpp
// Author      : abb
// Version     :
// Copyright   : [My copyright notice]
// Description : test program
//============================================================================

#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ull = unsigned long long;
using pll = pair<ll, ll>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vd = vector<double>;
using vvd = vector<vd>;
using vs = vector<string>;
using vvs = vector<vs>;

#define REP(i, m, n) for (ll i = (m), i##_len = (ll)(n); i < i##_len; ++(i))
#define REPM(i, m, n) for (ll i = (m), i##_max = (ll)(n); i <= i##_max; ++(i))
#define REPR(i, m, n) for (ll i = (m), i##_min = (ll)(n); i >= i##_min; --(i))
#define REPD(i, m, n, d)                                                       \
  for (ll i = (m), i##_len = (ll)(n); i < i##_len; i += (d))
#define REPMD(i, m, n, d)                                                      \
  for (ll i = (m), i##_max = (ll)(n); i <= i##_max; i += (d))
#define REPI(itr, ds) for (auto itr = ds.begin(); itr != ds.end(); itr++)
#define ALL(x) begin(x), end(x)
#define SIZE(x) ((ll)(x).size())

constexpr ll INF = 1'010'000'000'000'000'017LL;
constexpr ll MOD = 1'000'000'007LL; // 1e9 + 7
constexpr double EPS = 1e-12;
constexpr double PI = 3.14159265358979323846;

// vector入力
template <typename T> istream &operator>>(istream &is, vector<T> &vec) {
  for (T &x : vec)
    is >> x;
  return is;
}

// vector出力
template <typename T> ostream &operator<<(ostream &os, vector<T> &vec) {
  ll len = SIZE(vec);
  os << "{";
  for (int i = 0; i < len; i++)
    os << vec[i] << (i == len - 1 ? "" : ", ");
  os << "}";
  return os;
}

// pair出力
template <typename T, typename U>
ostream &operator<<(ostream &os, pair<T, U> &pair_var) {
  os << "(" << pair_var.first << ", " << pair_var.second << ")";
  return os;
}

// map出力
template <typename T, typename U>
ostream &operator<<(ostream &os, map<T, U> &map_var) {
  os << "{";
  REPI(itr, map_var) {
    os << *itr;
    itr++;
    if (itr != map_var.end())
      os << ", ";
    itr--;
  }
  os << "}";
  return os;
}

// set 出力
template <typename T> ostream &operator<<(ostream &os, set<T> &set_var) {
  os << "{";
  REPI(itr, set_var) {
    os << *itr;
    itr++;
    if (itr != set_var.end())
      os << ", ";
    itr--;
  }
  os << "}";
  return os;
}

// dump
#define DUMPOUT cerr
void dump_func() { DUMPOUT << endl; }
template <class Head, class... Tail>
void dump_func(Head &&head, Tail &&...tail) {
  DUMPOUT << head;
  if (sizeof...(Tail) > 0) {
    DUMPOUT << ", ";
  }
  dump_func(move(tail)...);
}

// chmax (更新「される」かもしれない値が前)
template <typename T, typename U, typename Comp = less<>>
bool chmax(T &xmax, const U &x, Comp comp = {}) {
  if (comp(xmax, x)) {
    xmax = x;
    return true;
  }
  return false;
}

// chmin (更新「される」かもしれない値が前)
template <typename T, typename U, typename Comp = less<>>
bool chmin(T &xmin, const U &x, Comp comp = {}) {
  if (comp(x, xmin)) {
    xmin = x;
    return true;
  }
  return false;
}

// container 内の最初の element のインデックスを探す
template <class Container>
ll indexof(const Container &container,
           const typename Container::value_type &element) {
  auto iter = find(container.begin(), container.end(), element);
  size_t index = distance(container.begin(), iter);
  if (index == container.size()) {
    index = -1;
  }
  return index;
}

// ローカル用
#define DEBUG_

#ifdef DEBUG_
#define DEB
#define dump(...)                                                              \
  DUMPOUT << "  " << string(#__VA_ARGS__) << ": "                              \
          << "[" << to_string(__LINE__) << ":" << __FUNCTION__ << "]" << endl  \
          << "    ",                                                           \
      dump_func(__VA_ARGS__)
#else
#define DEB if (false)
#define dump(...)
#endif

struct AtCoderInitialize {
  static constexpr int IOS_PREC = 15;
  static constexpr bool AUTOFLUSH = false;

  AtCoderInitialize() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    cout << fixed << setprecision(IOS_PREC);
    if (AUTOFLUSH)
      cout << unitbuf;
  }
} ATCODER_INITIALIZE;

/* #endregion */

/**
D - Gathering Children

実行時間制限: 2 sec / メモリ制限: 1024 MB

配点 : 400 点

問題文
マスの情報を表す、'L' と 'R' で構成された文字列 S が与えられます。
文字列 S の長さを N としたとき、N 個のマスが左右一列に並んでおり、左から i
番目のマスには S の左から i 番目の文字が書かれています。
ただし、左端のマスには必ず 'R'、右端のマスには必ず 'L' が書かれています。
はじめ、各マスには 1 人の子どもが居ます。
子どもたちはそれぞれ次の規則に従った移動を 10^{100} 回行います。

・今居るマスに書かれた文字に従って 1
マス移動する。すなわち、今居るマスに書かれた文字が 'L' のとき左隣のマスに、'R'
のとき右隣のマスに移動する。

10^{100} 回の移動の後に各マスに居る子どもの人数を求めてください。

制約

・S は長さ 2 以上 10^5 以下の文字列であり、S の各文字は 'L' または 'R' である。
・S の 1 文字目は 'R'、N 文字目は 'L' である。
 */
void solve() {
  string s;
  cin >> s;
  // 10^(100)回＝無限偶数回
  // 右隣がRになっているRなら0
  // 左隣がLになっているLなら0
  // RRRRL のように R が偶数個なら，[0,0,0,(R個数/2),(R個数/2)]が足される
  // RRRL のように R が奇数個なら，[0,0,(R個数/2+1),(R個数/2)]が足される
  // RLLLL のように L が偶数個なら，[(L個数/2),(L個数/2),0,0,0]が足される
  // RLLL のように L が奇数個なら，[(L個数/2),(L個数/2+1),0,0]が足される
  s.push_back('E');
  vll ret(SIZE(s), 0);
  char mode = 'R';
  ll length = 0;
  REP(i, 0, SIZE(s)) {
    if (s[i] == mode) {
      length++;
    } else {
      // mode の文字が length 個 あった
      if (mode == 'R') {
        ret[i] += length / 2;
        ret[i - 1] += length % 2 == 0 ? length / 2 : length / 2 + 1;
      } else {
        ret[i - length - 1] += length / 2;
        ret[i - length] += length % 2 == 0 ? length / 2 : length / 2 + 1;
      }
      dump(i, length, ret);
      mode = s[i];
      length = 1;
    }
  }
  dump(ret);
  REP(i, 0, SIZE(s) - 1) { cout << ret[i] << (i == SIZE(s) - 2 ? "" : " "); }
  cout << endl;
}

/**
 * エントリポイント．
 * @return 0.
 */
int main() {
  solve();
  return 0;
}


Fixed submission:
/* #region Head */

//============================================================================
// Name        : filename.cpp
// Author      : abb
// Version     :
// Copyright   : [My copyright notice]
// Description : test program
//============================================================================

#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ull = unsigned long long;
using pll = pair<ll, ll>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vd = vector<double>;
using vvd = vector<vd>;
using vs = vector<string>;
using vvs = vector<vs>;

#define REP(i, m, n) for (ll i = (m), i##_len = (ll)(n); i < i##_len; ++(i))
#define REPM(i, m, n) for (ll i = (m), i##_max = (ll)(n); i <= i##_max; ++(i))
#define REPR(i, m, n) for (ll i = (m), i##_min = (ll)(n); i >= i##_min; --(i))
#define REPD(i, m, n, d)                                                       \
  for (ll i = (m), i##_len = (ll)(n); i < i##_len; i += (d))
#define REPMD(i, m, n, d)                                                      \
  for (ll i = (m), i##_max = (ll)(n); i <= i##_max; i += (d))
#define REPI(itr, ds) for (auto itr = ds.begin(); itr != ds.end(); itr++)
#define ALL(x) begin(x), end(x)
#define SIZE(x) ((ll)(x).size())

constexpr ll INF = 1'010'000'000'000'000'017LL;
constexpr ll MOD = 1'000'000'007LL; // 1e9 + 7
constexpr double EPS = 1e-12;
constexpr double PI = 3.14159265358979323846;

// vector入力
template <typename T> istream &operator>>(istream &is, vector<T> &vec) {
  for (T &x : vec)
    is >> x;
  return is;
}

// vector出力
template <typename T> ostream &operator<<(ostream &os, vector<T> &vec) {
  ll len = SIZE(vec);
  os << "{";
  for (int i = 0; i < len; i++)
    os << vec[i] << (i == len - 1 ? "" : ", ");
  os << "}";
  return os;
}

// pair出力
template <typename T, typename U>
ostream &operator<<(ostream &os, pair<T, U> &pair_var) {
  os << "(" << pair_var.first << ", " << pair_var.second << ")";
  return os;
}

// map出力
template <typename T, typename U>
ostream &operator<<(ostream &os, map<T, U> &map_var) {
  os << "{";
  REPI(itr, map_var) {
    os << *itr;
    itr++;
    if (itr != map_var.end())
      os << ", ";
    itr--;
  }
  os << "}";
  return os;
}

// set 出力
template <typename T> ostream &operator<<(ostream &os, set<T> &set_var) {
  os << "{";
  REPI(itr, set_var) {
    os << *itr;
    itr++;
    if (itr != set_var.end())
      os << ", ";
    itr--;
  }
  os << "}";
  return os;
}

// dump
#define DUMPOUT cerr
void dump_func() { DUMPOUT << endl; }
template <class Head, class... Tail>
void dump_func(Head &&head, Tail &&...tail) {
  DUMPOUT << head;
  if (sizeof...(Tail) > 0) {
    DUMPOUT << ", ";
  }
  dump_func(move(tail)...);
}

// chmax (更新「される」かもしれない値が前)
template <typename T, typename U, typename Comp = less<>>
bool chmax(T &xmax, const U &x, Comp comp = {}) {
  if (comp(xmax, x)) {
    xmax = x;
    return true;
  }
  return false;
}

// chmin (更新「される」かもしれない値が前)
template <typename T, typename U, typename Comp = less<>>
bool chmin(T &xmin, const U &x, Comp comp = {}) {
  if (comp(x, xmin)) {
    xmin = x;
    return true;
  }
  return false;
}

// container 内の最初の element のインデックスを探す
template <class Container>
ll indexof(const Container &container,
           const typename Container::value_type &element) {
  auto iter = find(container.begin(), container.end(), element);
  size_t index = distance(container.begin(), iter);
  if (index == container.size()) {
    index = -1;
  }
  return index;
}

// ローカル用
// #define DEBUG_

#ifdef DEBUG_
#define DEB
#define dump(...)                                                              \
  DUMPOUT << "  " << string(#__VA_ARGS__) << ": "                              \
          << "[" << to_string(__LINE__) << ":" << __FUNCTION__ << "]" << endl  \
          << "    ",                                                           \
      dump_func(__VA_ARGS__)
#else
#define DEB if (false)
#define dump(...)
#endif

struct AtCoderInitialize {
  static constexpr int IOS_PREC = 15;
  static constexpr bool AUTOFLUSH = false;

  AtCoderInitialize() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    cout << fixed << setprecision(IOS_PREC);
    if (AUTOFLUSH)
      cout << unitbuf;
  }
} ATCODER_INITIALIZE;

/* #endregion */

/**
D - Gathering Children

実行時間制限: 2 sec / メモリ制限: 1024 MB

配点 : 400 点

問題文
マスの情報を表す、'L' と 'R' で構成された文字列 S が与えられます。
文字列 S の長さを N としたとき、N 個のマスが左右一列に並んでおり、左から i
番目のマスには S の左から i 番目の文字が書かれています。
ただし、左端のマスには必ず 'R'、右端のマスには必ず 'L' が書かれています。
はじめ、各マスには 1 人の子どもが居ます。
子どもたちはそれぞれ次の規則に従った移動を 10^{100} 回行います。

・今居るマスに書かれた文字に従って 1
マス移動する。すなわち、今居るマスに書かれた文字が 'L' のとき左隣のマスに、'R'
のとき右隣のマスに移動する。

10^{100} 回の移動の後に各マスに居る子どもの人数を求めてください。

制約

・S は長さ 2 以上 10^5 以下の文字列であり、S の各文字は 'L' または 'R' である。
・S の 1 文字目は 'R'、N 文字目は 'L' である。
 */
void solve() {
  string s;
  cin >> s;
  // 10^(100)回＝無限偶数回
  // 右隣がRになっているRなら0
  // 左隣がLになっているLなら0
  // RRRRL のように R が偶数個なら，[0,0,0,(R個数/2),(R個数/2)]が足される
  // RRRL のように R が奇数個なら，[0,0,(R個数/2+1),(R個数/2)]が足される
  // RLLLL のように L が偶数個なら，[(L個数/2),(L個数/2),0,0,0]が足される
  // RLLL のように L が奇数個なら，[(L個数/2),(L個数/2+1),0,0]が足される
  s.push_back('E');
  vll ret(SIZE(s), 0);
  char mode = 'R';
  ll length = 0;
  REP(i, 0, SIZE(s)) {
    if (s[i] == mode) {
      length++;
    } else {
      // mode の文字が length 個 あった
      if (mode == 'R') {
        ret[i] += length / 2;
        ret[i - 1] += length % 2 == 0 ? length / 2 : length / 2 + 1;
      } else {
        ret[i - length - 1] += length / 2;
        ret[i - length] += length % 2 == 0 ? length / 2 : length / 2 + 1;
      }
      dump(i, length, ret);
      mode = s[i];
      length = 1;
    }
  }
  dump(ret);
  REP(i, 0, SIZE(s) - 1) { cout << ret[i] << (i == SIZE(s) - 2 ? "" : " "); }
  cout << endl;
}

/**
 * エントリポイント．
 * @return 0.
 */
int main() {
  solve();
  return 0;
}