結果

問題 No.1666 累乗数
ユーザー rlangevinrlangevin
提出日時 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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 780 ms
250,864 KB
testcase_01 AC 821 ms
250,680 KB
testcase_02 AC 836 ms
250,700 KB
testcase_03 AC 822 ms
250,744 KB
testcase_04 AC 825 ms
251,100 KB
testcase_05 AC 822 ms
251,084 KB
testcase_06 AC 817 ms
251,184 KB
testcase_07 AC 822 ms
250,816 KB
testcase_08 AC 818 ms
251,036 KB
testcase_09 AC 831 ms
250,916 KB
testcase_10 AC 824 ms
250,764 KB
testcase_11 AC 825 ms
251,200 KB
testcase_12 AC 837 ms
250,680 KB
testcase_13 AC 843 ms
250,896 KB
testcase_14 AC 846 ms
251,080 KB
testcase_15 AC 832 ms
251,220 KB
testcase_16 AC 828 ms
250,868 KB
testcase_17 AC 833 ms
251,152 KB
testcase_18 AC 836 ms
250,976 KB
testcase_19 AC 828 ms
250,940 KB
権限があれば一括ダウンロードができます

ソースコード

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