Problem p02820 (C++)

Buggy submission (status: Runtime Error):
#include <algorithm>
#include <bitset>
#include <climits>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <math.h>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <stdio.h>
#include <string>
#include <utility>
#include <vector>

using namespace std;
typedef long long ll;
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}
// const ll INF = 1LL <<60;
// const int INF = 10000;

// 最大公約数
ll gcd(ll x, ll y) {
  ll tmp = 0;
  if (x < y) {
    tmp = x;
    x = y;
    y = tmp;
  }
  while (y > 0) {
    ll r = x % y;
    x = y;
    y = r;
  }
  return x;
}

// 最大公倍数
ll lcm(ll x, ll y) { return x / gcd(x, y) * y; }

const int MAX = 1e6 + 1;
const int MOD = 1e9 + 7;

long long fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算
long long COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

// 階乗
ll kaijo(ll k) {
  ll sum = 1;
  for (ll i = 1; i <= k; ++i) {
    sum *= i;
    sum %= 1000000000 + 7;
  }
  return sum;
}

// for(int i = ; i < ; i++){}

ll lmax(ll s, ll t) {
  if (s > t) {
    return s;
  } else {
    return t;
  }
}

ll lmin(ll s, ll t) {
  if (s < t) {
    return s;
  } else {
    return t;
  }
}

long long modpow(long long res, long long a, long long n, long long mod) {
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}

// sieve(MAX);でベクトルIsPrimeに「MAXまでの数値の素数の真偽」を格納する。
vector<bool> IsPrime;
void sieve(size_t max) {
  if (max + 1 > IsPrime.size()) {  // resizeで要素数が減らないように
    IsPrime.resize(max + 1, true); // IsPrimeに必要な要素数を確保
  }
  IsPrime[0] = false; // 0は素数ではない
  IsPrime[1] = false; // 1は素数ではない

  for (size_t i = 2; i * i <= max; ++i)     // 0からsqrt(max)まで調べる
    if (IsPrime[i])                         // iが素数ならば
      for (size_t j = 2; i * j <= max; ++j) // (max以下の)iの倍数は
        IsPrime[i * j] = false;             // 素数ではない
}
// ここから開始
int main() {
  int n, k, r, s, p;
  cin >> n >> k >> r >> s >> p;
  string t;
  cin >> t;
  vector<bool> rlist(n, false);
  vector<bool> slist(n, false);
  vector<bool> plist(n, false);
  for (int i = 0; i < t.length(); i++) {
    if (t[i] == 'r') {
      rlist[i] = true;
    }
    if (t[i] == 's') {
      slist[i] = true;
    } else if (t[i] == 'p') {
      plist[i] = true;
    }
  }
  int flg;
  ll sum = 0;
  for (int i = 0; i < n; i++) {
    if (rlist[i] == true && rlist[i + k] == true) {
      rlist[i + k] = false;
    }
    if (slist[i] == true && slist[i + k] == true) {
      slist[i + k] = false;
    }
    if (plist[i] == true && plist[i + k] == true) {
      plist[i + k] = false;
    }
  }
  for (int i = 0; i < n; i++) {
    if (rlist[i] == true) {
      // cout<<i<<endl;
      sum += p;
    }
    if (slist[i] == true) {
      // cout<<i<<endl;
      sum += r;
    }
    if (plist[i] == true) {
      // cout<<i<<endl;
      sum += s;
    }
  }
  // if(rlist.size()!=0){
  //   flg = rlist[0];
  // for(int i = 0; i < rlist.size(); i++){
  //   if(r)
  // }
  // if(slist.size()!=0){
  //       flg = slist[0];
  //       cout<<endl;
  //       for(int i = 0; i < slist.size(); i++){
  //         if(flg + k == slist[i]){
  //           cout<<"s"<<i<<" ";
  //         }
  //         else{
  //           sum+=r;
  //           flg = slist[i];
  //         }
  //     }
  //   }
  //   if(plist.size()!=0){
  //     flg = plist[0];
  //     cout<<endl;
  //     for(int i = 0; i < plist.size(); i++){
  //     cout<<plist[i]<<" ";
  //     if(flg + k == plist[i]){
  //       cout<<"p"<<i<<" ";
  //       ;
  //     }
  //     else{
  //       sum+=s;
  //       flg = plist[i];
  //     }
  //   }
  // }
  cout << sum << endl;
  return 0;
}


Fixed submission:
#include <algorithm>
#include <bitset>
#include <climits>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <math.h>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <stdio.h>
#include <string>
#include <utility>
#include <vector>

using namespace std;
typedef long long ll;
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> inline bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}
// const ll INF = 1LL <<60;
// const int INF = 10000;

// 最大公約数
ll gcd(ll x, ll y) {
  ll tmp = 0;
  if (x < y) {
    tmp = x;
    x = y;
    y = tmp;
  }
  while (y > 0) {
    ll r = x % y;
    x = y;
    y = r;
  }
  return x;
}

// 最大公倍数
ll lcm(ll x, ll y) { return x / gcd(x, y) * y; }

const int MAX = 1e6 + 1;
const int MOD = 1e9 + 7;

long long fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX; i++) {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
  }
}

// 二項係数計算
long long COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

// 階乗
ll kaijo(ll k) {
  ll sum = 1;
  for (ll i = 1; i <= k; ++i) {
    sum *= i;
    sum %= 1000000000 + 7;
  }
  return sum;
}

// for(int i = ; i < ; i++){}

ll lmax(ll s, ll t) {
  if (s > t) {
    return s;
  } else {
    return t;
  }
}

ll lmin(ll s, ll t) {
  if (s < t) {
    return s;
  } else {
    return t;
  }
}

long long modpow(long long res, long long a, long long n, long long mod) {
  while (n > 0) {
    if (n & 1)
      res = res * a % mod;
    a = a * a % mod;
    n >>= 1;
  }
  return res;
}

// sieve(MAX);でベクトルIsPrimeに「MAXまでの数値の素数の真偽」を格納する。
vector<bool> IsPrime;
void sieve(size_t max) {
  if (max + 1 > IsPrime.size()) {  // resizeで要素数が減らないように
    IsPrime.resize(max + 1, true); // IsPrimeに必要な要素数を確保
  }
  IsPrime[0] = false; // 0は素数ではない
  IsPrime[1] = false; // 1は素数ではない

  for (size_t i = 2; i * i <= max; ++i)     // 0からsqrt(max)まで調べる
    if (IsPrime[i])                         // iが素数ならば
      for (size_t j = 2; i * j <= max; ++j) // (max以下の)iの倍数は
        IsPrime[i * j] = false;             // 素数ではない
}
// ここから開始
int main() {
  int n, k, r, s, p;
  cin >> n >> k >> r >> s >> p;
  string t;
  cin >> t;
  vector<bool> rlist(n, false);
  vector<bool> slist(n, false);
  vector<bool> plist(n, false);
  for (int i = 0; i < t.length(); i++) {
    if (t[i] == 'r') {
      rlist[i] = true;
    }
    if (t[i] == 's') {
      slist[i] = true;
    } else if (t[i] == 'p') {
      plist[i] = true;
    }
  }
  int flg;
  ll sum = 0;
  for (int i = 0; i < n - k; i++) {
    if (rlist[i] == true && rlist[i + k] == true) {
      rlist[i + k] = false;
    }
    if (slist[i] == true && slist[i + k] == true) {
      slist[i + k] = false;
    }
    if (plist[i] == true && plist[i + k] == true) {
      plist[i + k] = false;
    }
  }
  for (int i = 0; i < n; i++) {
    if (rlist[i] == true) {
      // cout<<i<<endl;
      sum += p;
    }
    if (slist[i] == true) {
      // cout<<i<<endl;
      sum += r;
    }
    if (plist[i] == true) {
      // cout<<i<<endl;
      sum += s;
    }
  }
  // if(rlist.size()!=0){
  //   flg = rlist[0];
  // for(int i = 0; i < rlist.size(); i++){
  //   if(r)
  // }
  // if(slist.size()!=0){
  //       flg = slist[0];
  //       cout<<endl;
  //       for(int i = 0; i < slist.size(); i++){
  //         if(flg + k == slist[i]){
  //           cout<<"s"<<i<<" ";
  //         }
  //         else{
  //           sum+=r;
  //           flg = slist[i];
  //         }
  //     }
  //   }
  //   if(plist.size()!=0){
  //     flg = plist[0];
  //     cout<<endl;
  //     for(int i = 0; i < plist.size(); i++){
  //     cout<<plist[i]<<" ";
  //     if(flg + k == plist[i]){
  //       cout<<"p"<<i<<" ";
  //       ;
  //     }
  //     else{
  //       sum+=s;
  //       flg = plist[i];
  //     }
  //   }
  // }
  cout << sum << endl;
  return 0;
}