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

#include <bits/stdc++.h>
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define codefor                                                                \\
  int test;                                                                    \\
  scanf("%d", &test);                                                          \\
  while (test--)
#define yn(ans) cout << (ans ? "Yes" : "No") << endl
#define YN(ans) cout << (ans ? "YES" : "NO") << endl
#define umap unordered_map
#define uset unordered_set
using namespace std;
using ll = long long;
// 入力系
void scan(int &a) { scanf("%d", &a); }
void scan(long long &a) { scanf("%lld", &a); }
template <class T> void scan(T &a) { cin >> a; }
template <class T> void scan(vector<T> &vec) {
  for (auto &&it : vec)
    scan(it);
}
void in() {}
template <class Head, class... Tail> void in(Head &head, Tail &...tail) {
  scan(head);
  in(tail...);
}
// 出力系
void print(const int &a) { printf("%d", a); }
void print(const long long &a) { printf("%lld", a); }
void print(const double &a) { printf("%.15lf", a); }
template <class T> void print(const T &a) { cout << a; }
template <class T> void print(const vector<T> &vec) {
  if (vec.empty())
    return;
  print(vec[0]);
  for (auto it = vec.begin(); ++it != vec.end();) {
    putchar(' ');
    print(*it);
  }
}
void out() { putchar('\\n'); }
template <class T> void out(const T &t) {
  print(t);
  putchar('\\n');
}
template <class Head, class... Tail>
void out(const Head &head, const Tail &...tail) {
  print(head);
  putchar(' ');
  out(tail...);
}
// デバッグ系
template <class T> void dprint(const T &a) { cerr << a; }
template <class T> void dprint(const vector<T> &vec) {
  if (vec.empty())
    return;
  cerr << vec[0];
  for (auto it = vec.begin(); ++it != vec.end();) {
    cerr << " " << *it;
  }
}
void debug() { cerr << endl; }
template <class T> void debug(const T &t) {
  dprint(t);
  cerr << endl;
}
template <class Head, class... Tail>
void debug(const Head &head, const Tail &...tail) {
  dprint(head);
  cerr << " ";
  debug(tail...);
}

int main() {
  int n;
  in(n);
  int ans = 0, a;
  vector<int> vec(n, 0);
  for (int x = 1; x <= 100; x++) {
    for (int y = 1; y <= 100; y++) {
      for (int z = 1; z <= 100; z++) {
        a = x * x + y * y + z * z + x * y + y * z + z * x;
        if (a <= n) {
          vec[a - 1]++;
        }
      }
    }
  }
  rep(i, n) { out(vec[i]); }
}


Error: TLE

Fixed submission:

#include <bits/stdc++.h>
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define codefor                                                                \\
  int test;                                                                    \\
  scanf("%d", &test);                                                          \\
  while (test--)
#define yn(ans) cout << (ans ? "Yes" : "No") << endl
#define YN(ans) cout << (ans ? "YES" : "NO") << endl
#define umap unordered_map
#define uset unordered_set
using namespace std;
using ll = long long;
// 入力系
void scan(int &a) { scanf("%d", &a); }
void scan(long long &a) { scanf("%lld", &a); }
template <class T> void scan(T &a) { cin >> a; }
template <class T> void scan(vector<T> &vec) {
  for (auto &&it : vec)
    scan(it);
}
void in() {}
template <class Head, class... Tail> void in(Head &head, Tail &...tail) {
  scan(head);
  in(tail...);
}
// 出力系
void print(const int &a) { printf("%d", a); }
void print(const long long &a) { printf("%lld", a); }
void print(const double &a) { printf("%.15lf", a); }
template <class T> void print(const T &a) { cout << a; }
template <class T> void print(const vector<T> &vec) {
  if (vec.empty())
    return;
  print(vec[0]);
  for (auto it = vec.begin(); ++it != vec.end();) {
    putchar(' ');
    print(*it);
  }
}
void out() { putchar('\\n'); }
template <class T> void out(const T &t) {
  print(t);
  putchar('\\n');
}
template <class Head, class... Tail>
void out(const Head &head, const Tail &...tail) {
  print(head);
  putchar(' ');
  out(tail...);
}
// デバッグ系
template <class T> void dprint(const T &a) { cerr << a; }
template <class T> void dprint(const vector<T> &vec) {
  if (vec.empty())
    return;
  cerr << vec[0];
  for (auto it = vec.begin(); ++it != vec.end();) {
    cerr << " " << *it;
  }
}
void debug() { cerr << endl; }
template <class T> void debug(const T &t) {
  dprint(t);
  cerr << endl;
}
template <class Head, class... Tail>
void debug(const Head &head, const Tail &...tail) {
  dprint(head);
  cerr << " ";
  debug(tail...);
}

int main() {
  int n;
  in(n);
  int ans = 0;
  for (int i = 1; i <= n; i++) {
    ans = 0;
    for (int x = 1; x * x <= i; x++) {
      for (int y = 1; y * y <= i; y++) {
        if (x * x + x * y + y * y > n)
          break;
        for (int z = 1; z * z <= i; z++) {
          if (x * x + y * y + z * z + x * y + y * z + z * x > n)
            break;
          if (x * x + y * y + z * z + x * y + y * z + z * x == i)
            ans++;
        }
      }
    }
    printf("%d\\n", ans);
  }
}