Pinta-ala

#include <algorithm>
#include <array>
#include <iostream>
#include <vector>
#include <cmath>

using namespace std;

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

    for (int i = 0; i < n; i++) {
        int alkup, uusi;
        int a = 0;
        cin >> alkup >> uusi;

        if (alkup == 1) {
            cout << "YES" << endl;
            break;
        }

        //cout << "aaaa" << endl;

        for (int j = 1; j <= (int) sqrt(alkup); j++) {
            if (alkup % j == 0) {
                //cout << j << " " << alkup / j + 1 << endl;
                if ((j + 1) * (alkup / j + 1) == uusi) {
                    cout << "YES" << endl;
                    a = 1;
                    break;
                }
            }
        }

        if (a == 0) {
            cout << "NO" << endl;
        }
    }
}

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: WRONG ANSWER