結果
問題 |
No.1666 累乗数
|
ユーザー |
![]() |
提出日時 | 2023-09-06 12:27:32 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 846 ms / 2,000 ms |
コード長 | 975 bytes |
コンパイル時間 | 199 ms |
コンパイル使用メモリ | 82,384 KB |
実行使用メモリ | 251,220 KB |
最終ジャッジ日時 | 2024-06-24 07:05:16 |
合計ジャッジ時間 | 18,719 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 19 |
ソースコード
import sys input = sys.stdin.readline from bisect import * def isqrt(n): yes = 1 no = 10 ** 9 + 5 while no - yes != 1: mid = (yes + no)//2 if mid * mid <= n: yes = mid else: no = mid return yes inf = 10 ** 18 + 5 S = set() for n in range(3, 65): now = 2 while True: v = now ** n if v >= inf: break v2 = isqrt(v) ** 2 if v != v2: S.add(now ** n) now += 1 S = sorted(list(S)) T = int(input()) for _ in range(T): K = int(input()) yes = inf no = 0 while yes - no != 1: mid = (yes + no)//2 i2 = isqrt(mid) ans = -1 if i2 ** 2 == mid: ans = mid else: ans = (i2 + 1) ** 2 i3over = bisect_left(S, mid) if S[i3over] == mid: i3over += 1 if i2 + i3over >= K: yes = mid else: no = mid print(yes)