#859691 (Crystal) No.2271 平方根の１３桁精度近似計算

• 問題: No.2271 平方根の１３桁精度近似計算
• ユーザー: tomerun
• 提出日時: 2023-04-14 23:12:54
• 言語: Crystal (1.11.2)
• 結果: WA
• 実行時間: -
• コード長: 1,078 bytes
• コンパイル時間: 12,513 ms
• コンパイル使用メモリ: 296,568 KB
• 実行使用メモリ: 5,888 KB
• 最終ジャッジ日時: 2024-10-10 14:18:02
• 合計ジャッジ時間: 15,763 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	1 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	WA	-
testcase_12	WA	-
testcase_13	AC	2 ms; 5,248 KB
testcase_14	AC	2 ms; 5,248 KB
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	AC	3 ms; 5,248 KB
testcase_19	AC	9 ms; 5,248 KB
testcase_20	AC	2 ms; 5,248 KB
testcase_21	AC	2 ms; 5,248 KB
testcase_22	WA	-
testcase_23	WA	-
testcase_24	AC	2 ms; 5,248 KB
testcase_25	AC	2 ms; 5,248 KB
testcase_26	WA	-
testcase_27	AC	2 ms; 5,248 KB
testcase_28	AC	2 ms; 5,248 KB
testcase_29	WA	-
testcase_30	AC	4 ms; 5,248 KB
testcase_31	AC	1 ms; 5,248 KB
testcase_32	AC	2 ms; 5,248 KB
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	AC	10 ms; 5,888 KB
testcase_37	AC	1 ms; 5,248 KB
testcase_38	AC	2 ms; 5,248 KB
testcase_39	AC	2 ms; 5,248 KB

ソースコード

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
while ans > 1 << 29
  ans -= 5 ** 13
end
v = ans * ans - n
c = 0
while v != 0 && v % 5 == 0
  v //= 5
  c += 1
end
if v != 0 && c < e
  puts q.size
  puts q[0]
  puts ans
  puts c
  puts "invalid"
  while true
  end
end
puts ans