题解 | A×A=B

#include <iostream>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); // 加速输入输出
    int T;
    cin >> T;
    while (T--) {
        long long b;
        cin >> b;
        long long low = 0, high = 1000000000LL; // 设置二分查找上界为10^9
        bool found = false;
        while (low <= high) {
            long long mid = low + (high - low) / 2;
            long long square = mid * mid;
            if (square == b) {
                found = true;
                break;
            } else if (square < b) {
                low = mid + 1;
            } else {
                high = mid - 1;
            }
        }
        if (found) {
            cout << "YES\n";
        } else {
            cout << "NO\n";
        }
    }
    return 0;
}