結果

問題 No.1666 累乗数
ユーザー rlangevin
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0