結果

問題 No.1218 Something Like a Theorem
ユーザー takakin
提出日時 2020-09-09 21:53:40
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 339 bytes
コンパイル時間 120 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-12-16 09:16:39
合計ジャッジ時間 1,582 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=lambda: sys.stdin.readline().rstrip()
n,z=map(int,input().split())
if n==1:
	if z==1:
		print("No")
	else:
		print("Yes")
elif n>=3:
	print("No")
else:
	zz=int(z**0.5)+1
	A=[int(i)**2 for i in range(1,zz)]
	A=set(A)
	for x in range(1,zz):
		if z-x**2 in A:
			print("Yes")
			break
		else:
			continue
	else:
		print("No")
0