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;
            continue;
        }

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

        //cout << alkup << " -> " << uusi << endl;
        for (double j = 0.01; j <= sqrt(alkup) + 1; j++) {
            //cout << (j + 1) * (alkup / j + 1) - uusi << endl;
            int b = (j + 1) * (alkup / j + 1) - uusi;

            if (b == 0) {
                a = 2;
                break;
            }

            if (a == 0 && b > 0) {
                a = 1;
            }

            if (a == 0 && b > 0) {
                a = 3;
            }

            if (a == 1 && b < 0) {
                a = 2;
                break;
            }

            if (a == 3 && b < 0) {
                a = 2;
                break;
            }

            /*if ((j + 1) * (alkup / j + 1) - uusi) {
                cout << "YES" << endl;
                a = 2;
                break;
            }*/
        }

        if (a != 2) {
            cout << "NO" << endl;
        } else{
            cout << "YES" << endl;
        }
    }
}

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER