結果

問題 No.2271 平方根の13桁精度近似計算
ユーザー tomerun
提出日時 2023-04-14 23:08:01
言語 Crystal
(1.14.0)
結果
TLE  
実行時間 -
コード長 1,063 bytes
コンパイル時間 13,421 ms
コンパイル使用メモリ 296,508 KB
実行使用メモリ 13,312 KB
最終ジャッジ日時 2024-10-10 14:11:16
合計ジャッジ時間 15,884 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 5 TLE * 1 -- * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

n = read_line.to_i64
e = read_line.to_i
if e == 0
  puts 0
  exit
end
if n < 0
  n %= 5 ** e
  n += 5 ** e
end
ab = n.abs
a = Array.new(e + 1) do
  v = ab % 5
  ab //= 5
  n < 0 ? (5 - v) % 5 : v
end
if a[0] == 2 || a[0] == 3
  puts "NaN"
  exit
end
if a[0] == 0
  q = [[0, 0]]
elsif a[0] == 1
  q = [[1, 0], [4, 3]]
else
  q = [[2, 0], [3, 1]]
end
1.upto(e) do |i|
  nq = [] of Array(Int32)
  q.each do |cand|
    sum = cand[i]
    1.upto(i - 1) do |j|
      sum += cand[j] * cand[i - j]
    end
    5.times do |j|
      if (j * cand[0] * 2 + sum) % 5 == a[i]
        carry = (j * cand[0] * 2 + sum) // 5
        nc = cand.dup
        nc[i] = j
        nc << carry
        nq << nc
      end
    end
  end
  if nq.empty?
    puts "NaN"
    exit
  end
  q = nq
end
ans = 0i64
base = 1i64
(e + 1).times do |i|
  ans += base * q[0][i]
  base *= 5
end
if ans > 1 << 29
  raise RuntimeError.new
end
v = ans * ans - n
c = 0
while v % 5 == 0
  v //= 5
  c += 1
end
if c < e
  puts q.size
  puts q[0]
  puts ans
  puts c
  puts "invalid"
  while true
  end
end
puts ans
0