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

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

#define lli long long
#define all(i) i.begin(), i.end()
#define rall(i) i.rbegin(), i.rend()
#define rep0(i, j) for (int i = 0; i < j; i++)
#define rep1(i, j) for (int i = 1; i <= j; i++)
#define rep0d(i, j) for (int i = j - 1; i >= 0; i--)
#define MOD 1000000007
#define MAX (lli)1e12
typedef pair<lli, lli> p;
class Graph {
  int V;
  vector<list<p>> adj;
  const int INF = (int)1e6;

public:
  Graph(lli V) : adj(V) { this->V = V; }

  void addEdge(int u, int v, int w) { adj[u].push_back(make_pair(v, w)); }

  lli edgeNum() {
    lli sum = 0;
    rep0(i, V) sum += adj[i].size();
    return sum;
  }

  bool route(lli from, lli to) { // vec[i]==j: j->iっていうもとに戻れる道がある
    vector<lli> vec(V, -1);
    dfs(1, from, vec);
    return vec[to] >= 0;
  }

  void dfs(lli ed, lli num, vector<lli> &vec) {
    for (auto e : adj[num]) {
      if (vec[e.first] != -1)
        continue;
      vec[e.first] = num;
      dfs(ed + 1, e.first, vec);
    }
  }

  vector<int> bfs(int n) {
    queue<int> q;
    vector<int> vec(V, -1);

    q.push(n);
    vec[n] = 0;
    while (!q.empty()) {
      int p = q.front();
      q.pop();
      for (auto e : adj[p]) {
        if (vec[e.first] == -1) {
          q.push(e.first);
          vec[e.first] = e.second + vec[p];
        }
      }
    }
    return vec;
  }

  vector<int> shortestPath(int src) {
    priority_queue<p, vector<p>, greater<p>> pq;

    vector<int> dist(V, INF);
    pq.push(make_pair(0, src));
    dist[src] = 0;

    while (!pq.empty()) {
      int u = pq.top().second;
      pq.pop();

      list<p>::iterator i;
      for (auto i = adj[u].begin(); i != adj[u].end(); ++i) {
        int v = i->first;
        int weight = i->second;

        if (dist[v] > dist[u] + weight) {
          dist[v] = dist[u] + weight;
          pq.push(make_pair(dist[v], v));
        }
      }
    }
    for (int i = 0; i < V; i++)
      if (dist[i] == INF)
        dist[i] = -1;
    return dist;
  }
};

struct UF {
  vector<int> par, size;

  UF(int N) : par(N), size(N, 1) { rep0(i, N) par[i] = i; }

  int root(int x) {
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]);
  }

  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return;
    par[rx] = ry;
    size[rx] += size[ry];
  }

  bool same(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    return rx == ry;
  }

  int getSize(int x) { return size[x]; }
};
struct BIT {
  vector<lli> tab; // bit table
  BIT(int n) : tab(n + 1) { rep0(i, n + 1) tab[i] = 0; }

  void add(int p, lli num) {
    for (++p; p < tab.size(); p += p & -p)
      tab[p] += num;
  }

  lli sum(int p) { // from 0 to p
    lli ans = 0;
    for (++p; p > 0; p -= p & -p)
      ans += tab[p];
    return ans;
  }

  void debug() {
    rep0(i, tab.size()) cout << tab[i] << ' ';
    cout << endl;
  }
};

// pairに関するもの
lli sum(p c) { return (c.first + c.second); }
p operator-(const p &l, const p &r) {
  return {l.first - r.first, l.second - r.second};
}
bool cmpf(const pair<lli, lli> left, const pair<lli, lli> right) {
  return left.first < right.first;
}
bool cmps(const pair<lli, lli> left, const pair<lli, lli> right) {
  return left.second < right.second;
}

lli gcd(lli a, lli b) {
  if (a < b)
    return gcd(b, a);
  lli temp;
  while ((temp = a % b)) {
    a = b;
    b = temp;
  }
  return b;
}

struct nCr {
  vector<lli> f, rf;
  lli count = 0;
  nCr(lli num) : f(num), rf(num) {
    count = num;
    f[0] = 1;
    rep1(i, num) f[i] = (f[i - 1] * i) % MOD;
    rep0(i, num + 1) rf[i] = inv(f[i]);
  }

  lli inv(lli x) {
    lli res = 1;
    lli k = MOD - 2;
    lli y = x;
    while (k) {
      if (k & 1)
        res = (res * y) % MOD;
      y = (y * y) % MOD;
      k /= 2;
    }
    return res;
  }
  lli C(lli n, lli k) {
    if (k > n || k < 0)
      return 0;
    lli a = f[n];
    lli b = rf[n - k];
    lli c = rf[k];

    lli bc = (b * c) % MOD;
    return (a * bc) % MOD;
  }
};
int main() {
  lli n, k;
  cin >> n >> k;
  vector<lli> pm(n + 1), sc(n + 1);
  rep1(i, n) cin >> pm[i];
  rep1(i, n) cin >> sc[i];

  vector<vector<lli>> arr(n + 1, vector<lli>(n + 1, 0));

  lli ans = -MAX * MAX;
  rep1(i, n) {
    lli ct = k;
    lli j = pm[i], score = 0, l;
    arr[i][1] = sc[j];
    for (l = 1; i != j; l++) {
      j = pm[j];
      arr[i][l + 1] = arr[i][l] + sc[j];
    }
    if (k > l) {
      if (arr[i][l] > 0) {
        score += arr[i][l] * ((k - l) / l);
        ct %= l;
        lli tp = -MAX;
        rep0(m, l + 1) tp = max(arr[i][m], tp);
        rep0(m, ct + 1) tp = max(arr[i][m] + arr[i][l], tp);
        score += tp;
      } else {
        lli tp = -MAX;
        rep1(m, l + 1) tp = max(arr[i][m], tp);
        score += tp;
      }
    } else {
      lli tp = -MAX;
      rep1(m, ct) tp = max(arr[i][m], tp);
      score += tp;
    }
    // cout << i << ' ' << score << endl;
    ans = max(score, ans);
  }
  cout << ans;
}


Error: WA

Fixed submission:

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

#define lli long long
#define all(i) i.begin(), i.end()
#define rall(i) i.rbegin(), i.rend()
#define rep0(i, j) for (int i = 0; i < j; i++)
#define rep1(i, j) for (int i = 1; i <= j; i++)
#define rep0d(i, j) for (int i = j - 1; i >= 0; i--)
#define MOD 1000000007
#define MAX (lli)1e12
typedef pair<lli, lli> p;
class Graph {
  int V;
  vector<list<p>> adj;
  const int INF = (int)1e6;

public:
  Graph(lli V) : adj(V) { this->V = V; }

  void addEdge(int u, int v, int w) { adj[u].push_back(make_pair(v, w)); }

  lli edgeNum() {
    lli sum = 0;
    rep0(i, V) sum += adj[i].size();
    return sum;
  }

  bool route(lli from, lli to) { // vec[i]==j: j->iっていうもとに戻れる道がある
    vector<lli> vec(V, -1);
    dfs(1, from, vec);
    return vec[to] >= 0;
  }

  void dfs(lli ed, lli num, vector<lli> &vec) {
    for (auto e : adj[num]) {
      if (vec[e.first] != -1)
        continue;
      vec[e.first] = num;
      dfs(ed + 1, e.first, vec);
    }
  }

  vector<int> bfs(int n) {
    queue<int> q;
    vector<int> vec(V, -1);

    q.push(n);
    vec[n] = 0;
    while (!q.empty()) {
      int p = q.front();
      q.pop();
      for (auto e : adj[p]) {
        if (vec[e.first] == -1) {
          q.push(e.first);
          vec[e.first] = e.second + vec[p];
        }
      }
    }
    return vec;
  }

  vector<int> shortestPath(int src) {
    priority_queue<p, vector<p>, greater<p>> pq;

    vector<int> dist(V, INF);
    pq.push(make_pair(0, src));
    dist[src] = 0;

    while (!pq.empty()) {
      int u = pq.top().second;
      pq.pop();

      list<p>::iterator i;
      for (auto i = adj[u].begin(); i != adj[u].end(); ++i) {
        int v = i->first;
        int weight = i->second;

        if (dist[v] > dist[u] + weight) {
          dist[v] = dist[u] + weight;
          pq.push(make_pair(dist[v], v));
        }
      }
    }
    for (int i = 0; i < V; i++)
      if (dist[i] == INF)
        dist[i] = -1;
    return dist;
  }
};

struct UF {
  vector<int> par, size;

  UF(int N) : par(N), size(N, 1) { rep0(i, N) par[i] = i; }

  int root(int x) {
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]);
  }

  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return;
    par[rx] = ry;
    size[rx] += size[ry];
  }

  bool same(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    return rx == ry;
  }

  int getSize(int x) { return size[x]; }
};
struct BIT {
  vector<lli> tab; // bit table
  BIT(int n) : tab(n + 1) { rep0(i, n + 1) tab[i] = 0; }

  void add(int p, lli num) {
    for (++p; p < tab.size(); p += p & -p)
      tab[p] += num;
  }

  lli sum(int p) { // from 0 to p
    lli ans = 0;
    for (++p; p > 0; p -= p & -p)
      ans += tab[p];
    return ans;
  }

  void debug() {
    rep0(i, tab.size()) cout << tab[i] << ' ';
    cout << endl;
  }
};

// pairに関するもの
lli sum(p c) { return (c.first + c.second); }
p operator-(const p &l, const p &r) {
  return {l.first - r.first, l.second - r.second};
}
bool cmpf(const pair<lli, lli> left, const pair<lli, lli> right) {
  return left.first < right.first;
}
bool cmps(const pair<lli, lli> left, const pair<lli, lli> right) {
  return left.second < right.second;
}

lli gcd(lli a, lli b) {
  if (a < b)
    return gcd(b, a);
  lli temp;
  while ((temp = a % b)) {
    a = b;
    b = temp;
  }
  return b;
}

struct nCr {
  vector<lli> f, rf;
  lli count = 0;
  nCr(lli num) : f(num), rf(num) {
    count = num;
    f[0] = 1;
    rep1(i, num) f[i] = (f[i - 1] * i) % MOD;
    rep0(i, num + 1) rf[i] = inv(f[i]);
  }

  lli inv(lli x) {
    lli res = 1;
    lli k = MOD - 2;
    lli y = x;
    while (k) {
      if (k & 1)
        res = (res * y) % MOD;
      y = (y * y) % MOD;
      k /= 2;
    }
    return res;
  }
  lli C(lli n, lli k) {
    if (k > n || k < 0)
      return 0;
    lli a = f[n];
    lli b = rf[n - k];
    lli c = rf[k];

    lli bc = (b * c) % MOD;
    return (a * bc) % MOD;
  }
};
int main() {
  lli n, k;
  cin >> n >> k;
  vector<lli> pm(n + 1), sc(n + 1);
  rep1(i, n) cin >> pm[i];
  rep1(i, n) cin >> sc[i];

  vector<vector<lli>> arr(n + 1, vector<lli>(n + 1, 0));

  lli ans = -MAX;
  rep1(i, n) {
    lli ct = k;
    lli j = pm[i], score = 0, l;
    arr[i][1] = sc[j];
    for (l = 1; i != j; l++) {
      j = pm[j];
      arr[i][l + 1] = arr[i][l] + sc[j];
    }
    if (k >= l) {
      if (arr[i][l] > 0) {
        score += arr[i][l] * (k / l);
        ct %= l;
        lli tp = -MAX;
        rep0(j, ct + 1) tp = max(arr[i][j], tp);
        score += tp;
      } else {
        lli tp = -MAX;
        rep1(j, ct) tp = max(arr[i][j], tp);
        score += tp;
      }
    } else {
      lli tp = -MAX;
      rep1(j, ct) tp = max(arr[i][j], tp);
      score += tp;
    }
    // cout << i << ' ' << score << endl;
    ans = max(score, ans);
  }
  cout << ans;
}