結果

問題 No.2074 Product is Square ?
ユーザー shobonvip
提出日時 2022-09-16 21:45:37
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,668 bytes
コンパイル時間 399 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 157,700 KB
最終ジャッジ日時 2024-12-21 18:55:45
合計ジャッジ時間 33,331 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 27 WA * 1 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

# https://qiita.com/Kiri8128/items/eca965fe86ea5f4cbb98
def gcd(a, b):
	while b: a, b = b, a % b
	return a
def isPrimeMR(n):
	d = n - 1
	d = d // (d & -d)
	L = [2]
	for a in L:
		t = d
		y = pow(a, t, n)
		if y == 1: continue
		while y != n - 1:
			y = (y * y) % n
			if y == 1 or t == n - 1: return 0
			t <<= 1
	return 1
def findFactorRho(n):
	m = 1 << n.bit_length() // 8
	for c in range(1, 99):
		f = lambda x: (x * x + c) % n
		y, r, q, g = 2, 1, 1, 1
		while g == 1:
			x = y
			for i in range(r):
				y = f(y)
			k = 0
			while k < r and g == 1:
				ys = y
				for i in range(min(m, r - k)):
					y = f(y)
					q = q * abs(x - y) % n
				g = gcd(q, n)
				k += m
			r <<= 1
		if g == n:
			g = 1
			while g == 1:
				ys = f(ys)
				g = gcd(abs(x - ys), n)
		if g < n:
			if isPrimeMR(g): return g
			elif isPrimeMR(n // g): return n // g
			return findFactorRho(g)
def primeFactor(n):
	i = 2
	ret = {}
	rhoFlg = 0
	while i*i <= n:
		k = 0
		while n % i == 0:
			n //= i
			k += 1
		if k: ret[i] = k
		i += 1 + i % 2
		if i == 101 and n >= 2 ** 20:
			while n > 1:
				if isPrimeMR(n):
					ret[n], n = 1, 1
				else:
					rhoFlg = 1
					j = findFactorRho(n)
					k = 0
					while n % j == 0:
						n //= j
						k += 1
					ret[j] = k

	if n > 1: ret[n] = 1
	if rhoFlg: ret = {x: ret[x] for x in sorted(ret)}
	return ret

from collections import defaultdict
T = int(input())
for _ in range(T):
	n = int(input())
	a = list(map(int,input().split()))
	d = defaultdict(int)
	for i in range(n):
		for j, cnt in primeFactor(a[i]).items():
			d[j] += cnt
	mode = 1
	for i, cnt in d.items():
		if cnt % 2 != 0:
			mode = 0
	if mode:
		print("Yes")
	else:
		print("No")
0