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

#include <algorithm>
#include <bitset>
#include <climits>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <iostream>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> i_i;

#define PI 3.141592653589793238462643383279
#define mod 1000000007LL
#define rep(i, n) for (i = 0; i < n; ++i)
#define rep1(i, n) for (i = 1; i < n; ++i)
#define per(i, n) for (i = n - 1; i > -1; --i)
#define int(x)                                                                 \\
  int x;                                                                       \\
  scanf("%d", &x)
#define int2(x, y)                                                             \\
  int x, y;                                                                    \\
  scanf("%d%d", &x, &y)
#define int3(x, y, z)                                                          \\
  int x, y, z;                                                                 \\
  scanf("%d%d%d", &x, &y, &z)
#define int4(v, x, y, z)                                                       \\
  int v, x, y, z;                                                              \\
  scanf("%d%d%d%d", &v, &x, &y, &z)
#define int5(v, w, x, y, z)                                                    \\
  int v, w, x, y, z;                                                           \\
  scanf("%d%d%d%d%d", &v, &w, &x, &y, &z)
#define ll2(x, y)                                                              \\
  ll x, y;                                                                     \\
  cin >> x >> y;
#define scn(n, a) rep(i, n) cin >> a[i]
#define sc2n(n, a, b) rep(i, n) cin >> a[i] >> b[i]
#define pri(x) cout << (x) << "\\n"
#define pri2(x, y) cout << (x) << " " << (y) << "\\n"
#define pri3(x, y, z) cout << (x) << " " << (y) << " " << (z) << "\\n"
#define pb push_back
#define mp make_pair
#define all(a) (a).begin(), (a).end()
#define endl "\\n"
#define kabe puts("---------------------------")
#define kara puts("")
#define debug(x) cout << " --- " << (x) << "\\n"
#define debug2(x, y) cout << " --- " << (x) << " " << (y) << "\\n"
#define debug3(x, y, z)                                                        \\
  cout << " --- " << (x) << " " << (y) << " " << (z) << "\\n"
#define X first
#define Y second
#define eps 0.0001
#define prid(x) printf("%.15lf\\n", x)

signed main(void) {
  int i, j;
  for (int testcase = 0; testcase >= 0; testcase++) {
    int(q);
    int cnt = 0;
    for (; q--;) {
      ll d;
      scanf("%lld", &d);
      d = d * 2 + 1;
      if ((d & 1) == 0) {
        ++cnt;
        continue;
      }
      ll t;
      for (t = 3; t * t <= d; t += 2) {
        if (d % t == 0)
          break;
      }
      if (t * t > d)
        ++cnt;
    }
    pri(cnt);

    /*/

    //*/
    break;
  }
  return 0;
}

Error: TLE

Fixed submission:

#include <algorithm>
#include <bitset>
#include <climits>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <iostream>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> i_i;

#define PI 3.141592653589793238462643383279
#define mod 1000000007LL
#define rep(i, n) for (i = 0; i < n; ++i)
#define rep1(i, n) for (i = 1; i < n; ++i)
#define per(i, n) for (i = n - 1; i > -1; --i)
#define int(x)                                                                 \\
  int x;                                                                       \\
  scanf("%d", &x)
#define int2(x, y)                                                             \\
  int x, y;                                                                    \\
  scanf("%d%d", &x, &y)
#define int3(x, y, z)                                                          \\
  int x, y, z;                                                                 \\
  scanf("%d%d%d", &x, &y, &z)
#define int4(v, x, y, z)                                                       \\
  int v, x, y, z;                                                              \\
  scanf("%d%d%d%d", &v, &x, &y, &z)
#define int5(v, w, x, y, z)                                                    \\
  int v, w, x, y, z;                                                           \\
  scanf("%d%d%d%d%d", &v, &w, &x, &y, &z)
#define ll2(x, y)                                                              \\
  ll x, y;                                                                     \\
  cin >> x >> y;
#define scn(n, a) rep(i, n) cin >> a[i]
#define sc2n(n, a, b) rep(i, n) cin >> a[i] >> b[i]
#define pri(x) cout << (x) << "\\n"
#define pri2(x, y) cout << (x) << " " << (y) << "\\n"
#define pri3(x, y, z) cout << (x) << " " << (y) << " " << (z) << "\\n"
#define pb push_back
#define mp make_pair
#define all(a) (a).begin(), (a).end()
#define endl "\\n"
#define kabe puts("---------------------------")
#define kara puts("")
#define debug(x) cout << " --- " << (x) << "\\n"
#define debug2(x, y) cout << " --- " << (x) << " " << (y) << "\\n"
#define debug3(x, y, z)                                                        \\
  cout << " --- " << (x) << " " << (y) << " " << (z) << "\\n"
#define X first
#define Y second
#define eps 0.0001
#define prid(x) printf("%.15lf\\n", x)

signed main(void) {
  int i, j;
  for (int testcase = 0; testcase >= 0; testcase++) {
    int(q);
    int cnt = 0;
    for (; q--;) {
      ll d;
      scanf("%lld", &d);
      d = d * 2 + 1;
      if ((d & 1) == 0) {
        ++cnt;
        continue;
      }
      for (i = 3; i * i <= d; i += 2)
        if (d % i == 0)
          break;
      if (i * i > d)
        ++cnt;
    }
    pri(cnt);

    /*/

    //*/
    break;
  }
  return 0;
}