杭电OJ 2040 亲和数 C++

#include <iostream>
using namespace std;

int test(int a) {
  int b = 0;
  for (int i = 2; i <= a; i++) {
    if (a % i == 0)
      b += a / i;
  }
  return b;
}

int main() {
  int n;
  cin >> n;
  int a, b;
  while (n--) {
    cin >> a >> b;
    int aa = test(a);
    int bb = test(b);
    if (aa == b && bb == a)
      cout << "YES" << endl;
    else
      cout << "NO" << endl;
  }
  return 0;
}