結果

問題 No.2271 平方根の１３桁精度近似計算
コンテスト
ユーザー shobonvip
提出日時 2023-04-14 23:30:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 38 ms / 2,000 ms
コード長 674 bytes
コンパイル時間 315 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 54,172 KB
最終ジャッジ日時 2024-10-10 14:40:35
合計ジャッジ時間 3,393 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

n = int(input())
e = int(input())

if e == 0 or n == 0:
	print(0)
	exit()

if (n % (5 ** e) == 0):
	print(0)
	exit()

ee = e
nn = n
basis = 1
while n % 25 == 0:
	n //= 25
	e -= 2
	basis *= 5

if not (n % 5 == 1 or n % 5 == 4):
	print("NaN")
	exit()

a = -1
for i in range(5):
	if (i * i - n) % 5 == 0:
		a = i
		break

assert a >= 0

for t in range(2, e+1):
	c = (a * a - n) // (5 ** (t-1))
	y = -1
	for yy in range(5):
		if (2 * a * yy + c) % 5 == 0:
			y = yy
			break
	assert y >= 0
	b = (a + 5 ** (t-1) * y) % (5 ** t)
	a = b

a *= basis
a %= (5 ** ee)
if a > 2 ** 29:
	a -= 5 ** ee

if a < -2 ** 29:
	print("NaN")
	exit()

print(a)
#print((a * a - nn) % (5 ** ee) == 0)
0