#798293 (C++17) No.2074 Product is Square ?

• 問題: No.2074 Product is Square ?
• ユーザー: noya2
• 提出日時: 2022-09-16 22:45:15
• 言語: C++17 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 517 ms / 2,000 ms
• コード長: 7,494 bytes
• コンパイル時間: 4,281 ms
• コンパイル使用メモリ: 259,328 KB
• 最終ジャッジ日時: 2025-02-07 10:19:26
• ジャッジサーバーID （参考情報）: judge2 / judge2

ファイルパターン	結果
sample	AC * 1
other	AC * 33

ソースコード

#line 1 "c.cpp"
#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for (int i = 0; i < int(n); ++i)
#define repp(i,n,m) for (int i = m; i < int(n); ++i)
#define repb(i,n) for (int i = int(n)-1; i >= 0; --i)
#define all(v) v.begin(),v.end()
using namespace std;
using namespace atcoder;
using ll = long long;
using ld = long double;
using P = pair<int, int>;
using PL = pair<long long, long long>;
using pdd = pair<long double, long double>;
using pil = pair<int,ll>;
using pli = pair<ll,int>;
using ppi = pair<P,int>;
using pip = pair<int,P>;
const int INF = 1001001007;
const long long mod1 = 1000000007LL;
const long long mod2 = 998244353LL;
const ll inf = 2e18;
const ld pi = 3.14159265358979323;
const ld eps = 1e-7;
template<typename T>void o(T a);
template<class T>istream &operator>>(istream &is,vector<T> &v){for(auto &e:v)is>>e;return is;}
template<typename T>bool range(T a,T b,T x){return (a<=x&&x<b);}
template<typename T>bool rrange(T a,T b,T c,T d,T x,T y){return (range(a,c,x)&&range(b,d,y));}
template<typename T>void rev(vector<T> &v){reverse(v.begin(),v.end());}
void revs(string &s) {reverse(s.begin(),s.end());}
template<typename T>void sor(vector<T> &v, int f=0){sort(v.begin(),v.end());if(f!=0) rev(v);}
template<typename T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
template<typename T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<typename T>void uniq(vector<T> &v){sor(v);v.erase(unique(v.begin(),v.end()),v.end());}
template<typename T>T cel(T a,T b){return (a+b-1)/b;}
template<typename T1, typename T2>void print(pair<T1,T2> a);
template<typename T>void print(vector<T> v);
template<typename T>void print(vector<vector<T>> v);
void print(){ putchar(' '); }
void print(bool a){ printf("%d", a); }
void print(int a){ printf("%d", a); }
void print(long a){ printf("%ld", a); }
void print(long long a){ printf("%lld", a); }
void print(char a){ printf("%c", a); }
void print(char a[]){ printf("%s", a); }
void print(const char a[]){ printf("%s", a); }
void print(long double a){ printf("%.15Lf", a); }
void print(const string& a){ for(auto&& i : a) print(i); }
void print(unsigned int a){ printf("%u", a); }
template<class T> void print(const T& a){ cout << a; }
int out(){ putchar('\n'); return 0; }
template<class T> int out(const T& t){ print(t); putchar('\n'); return 0; }
template<class Head, class... Tail> int out(const Head& head, const Tail&... tail){ print(head); putchar(' '); out(tail...); return 0; }
void o(){cout<<"!?"<<endl;}
template<typename T>void o(T a){cout<<a<<endl;}
template<typename T1,typename T2>void print(pair<T1,T2> a){print(a.first);print(),print(a.second);}
template<typename T>void print(vector<T> v){for(auto ite=v.begin();ite!=v.end();){print(*ite);if(++ite!=v.end())print();}}
template<typename T>void print(vector<vector<T>> v){for(auto ite=v.begin();ite!=v.end();){print(*ite);if(++ite!=v.end())out();}}
void yes(){out("Yes");}
void no (){out("No");}
void yn (bool t){if(t)yes();else no();}
vector<int> dx = {0,1,0,-1,1,1,-1,-1};
vector<int> dy = {1,0,-1,0,1,-1,-1,1};
const string ALP = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const string alp = "abcdefghijklmnopqrstuvwxyz";
const string NUM = "0123456789";

void fast_io(){cin.tie(0); ios::sync_with_stdio(0); cout<<fixed<<setprecision(20);}

using mint = modint998244353;
void out(mint a){out(a.val());}
void out(vector<mint> a){vector<ll> b(a.size()); rep(i,a.size()) b[i] = a[i].val(); out(b);}
void out(vector<vector<mint>> a){for (auto v : a) out(v);}
istream &operator>>(istream &is,vector<mint> &v){for(auto &e:v){ll _x;is>>_x;e=_x;}return is;}

#line 2 "math.hpp"

namespace noya2{

using ll = long long;
ll safe_mod(ll a, ll m){ // m >= 1, 0 <= safe_mod(a,m) < m, safe_mod(a,m) = a (mod m)
    ll res = a % m;
    if (res < 0) res += m;
    return res;
} 

struct math{
    static ll gcd(ll a, ll b){return b?gcd(b,a%b):a;}
    static void ext_gcd1_plus(ll a, ll b, ll &x, ll &y){ // gcd(a,b) = 1, a >= 0, b >= 0, ax + by = 1
        if (b == 0){ // a = 1
            x = 1, y = 0;
            return ;
        }
        ext_gcd1_plus(b, a%b, y, x);
        x = safe_mod(x,b);
        y = (1 - a * x) / b;
    }
    static void ext_gcd1_minus(ll a, ll b, ll &x, ll &y){ // gcd(a,b) = 1, a >= 0, b >= 0, ax - by = 1
        if (b == 0){ // a = 1
            x = 1, y = 0;
            return ;
        }
        ext_gcd1_plus(b, a%b, y, x);
        x = safe_mod(x,b);
        y = (a * x - 1) / b;
    }
};

ll safe_gcd(ll a, ll b){return math::gcd(abs(a),abs(b));} // safe_gcd(0,0) = 0
ll safe_lcm(ll a, ll b){return abs(a) / safe_gcd(a,b) * abs(b);}

struct line{
    ll a, b, c; // ax + by = c
    void norm(){ // a >= 0, if (a == 0) -> b >= 0, gcd(a,b,c) = 1
        if (a < 0) a = -a, b = -b, c = -c;
        if (a == 0) if (b < 0) b = -b, c = -c;
        ll g = safe_gcd(safe_gcd(a,b),c);
        a /= g, b /= g, c /= g;
    }
    line (ll _a = 1, ll _b = 0, ll _c = 0) : a(_a), b(_b), c(_c) {
        assert(_a != 0 || _b != 0);
        norm();
    }
    line (ll x1, ll y1, ll x2, ll y2){ // (x1,y1) --- (x2,y2) 
        assert(x1 != x2 || y1 != y2);
        ll wx = x2 - x1, wy = y2 - y1;
        ll wg = safe_gcd(wx,wy);
        a = wy / wg, b = -wx / wg;
        c = a * x1 + b * y1;
        norm();
    }
    static line get_line(ll x1, ll y1, ll ex, ll ey){ // y = ey/ex (x - x1) + y1 
        return line(ey,-ex,ey*x1-ex*y1);
    }
    static line vertical_line(ll x1, ll y1, ll x2, ll y2){ // perpendicular bisector
        ll ex = y2 - y1, ey = x1 - x2;
        return line(ey*2,-ex*2,ey*(x1+x2)-ex*(y1+y2));
    }
    bool operator<(const line &rhs) const {
        return vector<ll>{a,b,c} < vector<ll>{rhs.a,rhs.b,rhs.c}; // to use set or map for line
    }
};

pair<ll,ll> ext_gcd(ll a, ll b, ll c){ // ax + by = c, |a|,|b|,|c| < 1e9, x <= max(|B|,|C|), y<= max(|A|,|C|)
    if (a < 0) return ext_gcd(-a,-b,-c);
    if (c < 0){
        pair<ll,ll> res = ext_gcd(a,b,-c);
        res.first = -res.first, res.second = -res.second;
        return res;
    }
    if (c == 0) return pair<ll,ll>(0,0);
    if (a == 0 && b == 0) return pair<ll,ll>(0,0); // answer not exist
    ll g = safe_gcd(a,b);
    if (c % g != 0) return pair<ll,ll>(0,0); // answer not exist
    a /= g, b /= g, c /= g;
    ll x, y;
    if (b == 0) return pair<ll,ll>(c,0);
    if (b > 0) math::ext_gcd1_plus(a,b,x,y);
    else math::ext_gcd1_minus(a,-b,x,y);
    x = safe_mod(x*c,abs(b));
    y = (c - a * x) / b;
    return pair<ll,ll>(x,y);
}

template<typename T>
T safe_ceil(T p, T q);

template<typename T>
T safe_floor(T p, T q){
    if (q < T(0)) return safe_floor(-p,-q);
    if (p >= T(0)) return p / q;
    return -safe_ceil(-p,q);
}

template<typename T>
T safe_ceil(T p, T q){
    if (q < T(0)) return safe_ceil(-p,-q);
    if (p >= T(0)) return (p + q - 1) / q;
    return -safe_floor(-p,q);
}

}// namespace noya2
#line 76 "c.cpp"

void solve(){
    int n; cin >> n;
    vector<ll> a(n); cin >> a;
    repp(i,n,1) rep(j,i){
        ll g = noya2::safe_gcd(a[i],a[j]);
        a[i] /= g; a[j] /= g;
    }
    auto is_square = [](ll x){
        ll le = 0, ri = 1'000'000'001LL;
        while (ri - le > 1){
            ll mid = (le + ri) / 2;
            if (mid * mid <= x) le = mid;
            else ri = mid;
        }
        return le * le == x;
    };
    bool ok = true;
    rep(i,n) if (!is_square(a[i])) ok = false;
    yn(ok);
}

int main(){
    fast_io();
    int t = 1; cin >> t;
    while(t--) solve();
}