結果

問題 No.2074 Product is Square ?
コンテスト
ユーザー taiga0629kyopro
提出日時 2022-09-16 21:47:14
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,530 bytes
コンパイル時間 140 ms
コンパイル使用メモリ 81,976 KB
実行使用メモリ 157,880 KB
最終ジャッジ日時 2024-12-21 18:59:26
合計ジャッジ時間 51,426 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 24 WA * 4 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 




from math import gcd
import random
from collections import defaultdict
def is_prime(n):
    if n==2:
        return True
    if n==1 or n&1==0:
        return False
    d=(n-1)>>1
    while d&1==0:
        d>>=1

    for k in range(100):
        a=random.randint(1,n-1)
        t=d
        y=pow(a, t, n)
        while t!=n-1 and y!=1 and y!=n-1:
            y=(y*y)%n
            t<<=1
        if y!=n-1 and t&1==0:
            return False
    return True
def prime_f(x,N):
    return (x*x+1)%N
def prime_fact(N):
    res=defaultdict(int)
    if N==1:
        return res
    p=2
    while(p<=10**4 and N>1):
        if N%p==0:
            while(N%p==0):
                res[p]+=1
                N//=p
        p+=1
    while(N>1):
        if is_prime(N):
            res[N]+=1
            break
        x=random.randrange(N)
        y=prime_f(x,N)
        i=1
        while(True):
            d=gcd(abs(x-y),N)
            if d==1:
                i+=1
            elif d==N:
                res[N]+=1
                return res
            else:
                res[d]+=1
                N//=d
                break
            x=prime_f(x,N)
            y=prime_f(prime_f(y,N),N)
    return res


T=int(input())
from collections import defaultdict
for iii in range(T):
    n=int(input())
    a=list(map(int,input().split()))
    ans=defaultdict(int)
    for x in a:
        pri=prime_fact(x)
        for p in pri:ans[p]+=pri[p]
    flag=1
    for p in ans:
        if ans[p]%2==1:flag=0
    print("Yes" if flag else "No")
0