Problem p03711 (C++)

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

#define int long long
#define pb push_back
#define ub upper_bound
#define lb upper_bound
#define mp make_pair
#define F first
#define S second
#define FOR(i, a, b) for (int(i) = (a); (i) < (b); (i)++)
#define REP(i, n) FOR(i, 0, n)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define SORT(c) sort((c).begin(), (c).end())
#define ve vector
#define vi vector<int>
#define vp vector<pair<int, int>>
#define vvi vector<vector<int>>

typedef long long ll;
const ll INF = LLONG_MAX - 100;
const ll mod = 1e9 + 7;
const int MAX_N = 2e5 + 5;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
vector<ll> prime;

int fac[MAX_N], inv[MAX_N];

template <class T = ll> T in() {
  T x;
  cin >> x;
  return (x);
}
void DEBUG(vector<int> a) {
  for (int i = 0; i < a.size(); i++)
    cout << a[i] << " ";
  cout << endl;
}
void EMP(int x) { cout << "!!!" << x << "!!!" << endl; }
ll GCD(ll a, ll b) {
  ll c;
  while (b != 0) {
    c = a % b;
    a = b;
    b = c;
  }
  return a;
}
ll LCM(ll a, ll b) { return a * b / GCD(a, b); }
ll POW(ll a, ll b, bool usemod = true) {
  ll c = 1LL;
  while (b > 0) {
    if (b & 1LL) {
      if (!usemod)
        c *= a;
      else
        c = a * c % mod;
    }
    if (!usemod)
      a *= a;
    else
      a = a * a % mod;
    b >>= 1LL;
  }
  return c;
}
void _nCr() {
  fac[0] = 1LL;
  for (int i = 1LL; i < MAX_N; i++) {
    fac[i] = fac[i - 1LL] * i % mod;
  }
  for (int i = 0; i < MAX_N; i++) {
    inv[i] = POW(fac[i], mod - 2);
  }
}
ll nCr(ll n, ll r) { return (fac[n] * inv[r] % mod) * inv[n - r] % mod; }
void PRI(ll n) {
  bool a[n + 1LL];
  for (int i = 0; i < n + 1LL; i++) {
    a[i] = 1LL;
  }
  for (int i = 2; i < n + 1LL; i++) {
    if (a[i]) {
      prime.pb(i);
      ll b = i;
      while (b <= n) {
        a[b] = 0;
        b += i;
      }
    }
  }
}
template <typename T> T chmin(T &a, T b) {
  if (a > b)
    a = b;
  return a;
}
template <typename T> T chmax(T &a, T b) {
  if (a < b)
    a = b;
  return b;
}

class Warshall_Floyd {
public:
  vector<vector<int>> d;
  int nodecnt = -1;

  Warshall_Floyd(int n) {
    d = vector<vector<int>>(n, vector<int>(n, INF));
    nodecnt = n;
  }

  void add(int s, int e, int cost) { d[s][e] = cost; }

  bool wf() {
    for (int k = 0; k < nodecnt; k++) {
      for (int i = 0; i < nodecnt; i++) {
        for (int j = 0; j < nodecnt; j++) {
          // k経由でi->j
          d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
        }
      }
    }
  }
};

bool solve() {
  set<int> A[3];
  A[0] = set<int>{1, 3, 5, 7, 8, 10, 12};
  A[2] = set<int>{4, 6, 9, 11};
  A[3] = set<int>{2};
  int x, y;
  cin >> x >> y;
  REP(i, 3) {
    if (A[i].find(x) != A[i].end() && A[i].find(y) != A[i].end()) {
      cout << "Yes" << endl;
      return 0;
    }
  }
  cout << "No" << endl;
}

signed main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  solve();
}

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

#define int long long
#define pb push_back
#define ub upper_bound
#define lb upper_bound
#define mp make_pair
#define F first
#define S second
#define FOR(i, a, b) for (int(i) = (a); (i) < (b); (i)++)
#define REP(i, n) FOR(i, 0, n)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define SORT(c) sort((c).begin(), (c).end())
#define ve vector
#define vi vector<int>
#define vp vector<pair<int, int>>
#define vvi vector<vector<int>>

typedef long long ll;
const ll INF = LLONG_MAX - 100;
const ll mod = 1e9 + 7;
const int MAX_N = 2e5 + 5;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
vector<ll> prime;

int fac[MAX_N], inv[MAX_N];

template <class T = ll> T in() {
  T x;
  cin >> x;
  return (x);
}
void DEBUG(vector<int> a) {
  for (int i = 0; i < a.size(); i++)
    cout << a[i] << " ";
  cout << endl;
}
void EMP(int x) { cout << "!!!" << x << "!!!" << endl; }
ll GCD(ll a, ll b) {
  ll c;
  while (b != 0) {
    c = a % b;
    a = b;
    b = c;
  }
  return a;
}
ll LCM(ll a, ll b) { return a * b / GCD(a, b); }
ll POW(ll a, ll b, bool usemod = true) {
  ll c = 1LL;
  while (b > 0) {
    if (b & 1LL) {
      if (!usemod)
        c *= a;
      else
        c = a * c % mod;
    }
    if (!usemod)
      a *= a;
    else
      a = a * a % mod;
    b >>= 1LL;
  }
  return c;
}
void _nCr() {
  fac[0] = 1LL;
  for (int i = 1LL; i < MAX_N; i++) {
    fac[i] = fac[i - 1LL] * i % mod;
  }
  for (int i = 0; i < MAX_N; i++) {
    inv[i] = POW(fac[i], mod - 2);
  }
}
ll nCr(ll n, ll r) { return (fac[n] * inv[r] % mod) * inv[n - r] % mod; }
void PRI(ll n) {
  bool a[n + 1LL];
  for (int i = 0; i < n + 1LL; i++) {
    a[i] = 1LL;
  }
  for (int i = 2; i < n + 1LL; i++) {
    if (a[i]) {
      prime.pb(i);
      ll b = i;
      while (b <= n) {
        a[b] = 0;
        b += i;
      }
    }
  }
}
template <typename T> T chmin(T &a, T b) {
  if (a > b)
    a = b;
  return a;
}
template <typename T> T chmax(T &a, T b) {
  if (a < b)
    a = b;
  return b;
}

class Warshall_Floyd {
public:
  vector<vector<int>> d;
  int nodecnt = -1;

  Warshall_Floyd(int n) {
    d = vector<vector<int>>(n, vector<int>(n, INF));
    nodecnt = n;
  }

  void add(int s, int e, int cost) { d[s][e] = cost; }

  bool wf() {
    for (int k = 0; k < nodecnt; k++) {
      for (int i = 0; i < nodecnt; i++) {
        for (int j = 0; j < nodecnt; j++) {
          // k経由でi->j
          d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
        }
      }
    }
  }
};

bool solve() {
  set<int> A[3];
  A[0] = set<int>{1, 3, 5, 7, 8, 10, 12};
  A[1] = set<int>{4, 6, 9, 11};
  A[2] = set<int>{2};
  int x, y;
  cin >> x >> y;
  REP(i, 3) {
    if (A[i].find(x) != A[i].end() && A[i].find(y) != A[i].end()) {
      cout << "Yes" << endl;
      return 0;
    }
  }
  cout << "No" << endl;
}

signed main() {
  cin.tie(0);
  ios::sync_with_stdio(false);
  solve();
}