Problem p03347 (C++)

Buggy submission (status: Time Limit Exceeded):
#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define fw(p) for (int w = 0; w < (p); w++)
#define fx(p) for (int x = 0; x < (p); x++)
#define fy(p) for (int y = 0; y < (p); y++)
#define fz(p) for (int z = 0; z < (p); z++)
#define fyg(p, g) for (int y = (g); y < (p); y++)
#define fzg(p, g) for (int z = (g); z < (p); z++)
#define ce(d) cout << d << endl;
#define vecp(p)                                                                \
  int aa;                                                                      \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecpl(p)                                                               \
  long long aa;                                                                \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecps(p)                                                               \
  string aa;                                                                   \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecp2(p)                                                               \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecpl2(p)                                                              \
  long long a b;                                                               \
  cin >> ab;                                                                   \
  (p).push_back(ab);
#define vecps2(p)                                                              \
  string ab;                                                                   \
  cin >> ab;                                                                   \
  (p).push_back(ab);
#define sorts(c) sort((c).begin(), (c).end());
#define reverses(c) reverse((c).begin(), (c).end());
#define vec(b) vector<int>(b);
#define vecl(b) vector<long long>(b);
#define vecs(b) vector<string>(b);
#define pb(b, a) (b).push_back((a));
#define doublece(a, b) cout << (a) << ' ' << (b) << endl;
#define pairs(s) vector<pair<int, int>>(s);
#define pairsp(s)                                                              \
  int aa, bb;                                                                  \
  cin >> aa >> bb;                                                             \
  (s).push_back(make_pair(aa, bb));
#define MOD 1000000007
#define cey ce("Yes")
#define cen ce("No")
#define ceY ce("YES")
#define ceN ce("NO")

int main() {
  int N;
  cin >> N;

  vec(P);
  fx(N) { vecp(P); }
  if (P[0] != 0) {
    ce(-1);
    return 0;
  }
  fx(N - 1) {
    if (P[x + 1] - P[x] > 1) {
      ce(-1);
      return 0;
    }
  }

  ll ans = 0;
  int i = 0;
  while (i < N) {
    int j = i;
    map<int, ll> K;
    while (i + 1 < N && P[i + 1] >= P[i]) {
      K[P[i]]++;
      i++;
    }
    ans += (K[P[i]] + 1) * P[i];
    for (int k = P[i] - 1; k > 0; k--) {
      ans += max(0LL, (K[k] - 1) * k);
    }
    i++;
  }
  ce(ans) return 0;
}


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

#define ll long long
#define fw(p) for (int w = 0; w < (p); w++)
#define fx(p) for (int x = 0; x < (p); x++)
#define fy(p) for (int y = 0; y < (p); y++)
#define fz(p) for (int z = 0; z < (p); z++)
#define fyg(p, g) for (int y = (g); y < (p); y++)
#define fzg(p, g) for (int z = (g); z < (p); z++)
#define ce(d) cout << d << endl;
#define vecp(p)                                                                \
  int aa;                                                                      \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecpl(p)                                                               \
  long long aa;                                                                \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecps(p)                                                               \
  string aa;                                                                   \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecp2(p)                                                               \
  cin >> aa;                                                                   \
  (p).push_back(aa);
#define vecpl2(p)                                                              \
  long long a b;                                                               \
  cin >> ab;                                                                   \
  (p).push_back(ab);
#define vecps2(p)                                                              \
  string ab;                                                                   \
  cin >> ab;                                                                   \
  (p).push_back(ab);
#define sorts(c) sort((c).begin(), (c).end());
#define reverses(c) reverse((c).begin(), (c).end());
#define vec(b) vector<int>(b);
#define vecl(b) vector<long long>(b);
#define vecs(b) vector<string>(b);
#define pb(b, a) (b).push_back((a));
#define doublece(a, b) cout << (a) << ' ' << (b) << endl;
#define pairs(s) vector<pair<int, int>>(s);
#define pairsp(s)                                                              \
  int aa, bb;                                                                  \
  cin >> aa >> bb;                                                             \
  (s).push_back(make_pair(aa, bb));
#define MOD 1000000007
#define cey ce("Yes")
#define cen ce("No")
#define ceY ce("YES")
#define ceN ce("NO")

int main() {
  int N;
  cin >> N;

  vec(P);
  fx(N) { vecp(P); }
  if (P[0] != 0) {
    ce(-1);
    return 0;
  }
  fx(N - 1) {
    if (P[x + 1] - P[x] > 1) {
      ce(-1);
      return 0;
    }
  }

  ll ans = 0;
  int i = 0;
  while (i < N) {
    int j = i;
    map<int, ll> K;
    while (i + 1 < N && P[i + 1] >= P[i]) {
      K[P[i]]++;
      i++;
    }
    ans += (K[P[i]] + 1) * P[i];
    for (int k = P[i] - 1; k >= P[j]; k--) {
      ans += max(0LL, (K[k] - 1) * k);
    }
    i++;
  }
  ce(ans) return 0;
}