結果

問題 No.3156 Count That Day's N
ユーザー norioc
提出日時 2025-06-25 17:31:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 795 ms / 3,000 ms
コード長 756 bytes
コンパイル時間 641 ms
コンパイル使用メモリ 82,224 KB
実行使用メモリ 255,128 KB
最終ジャッジ日時 2025-06-25 17:32:04
合計ジャッジ時間 17,254 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import isqrt

K, N = map(int, input().split())

x6list = []
for i in range(1, N+1):
    x6 = i ** 6
    if x6 > N: break
    x6list.append(x6)

x4list = []
for i in range(1, N+1):
    x4 = i ** 4
    if x4 > N: break
    x4list.append(x4)


def isq(n):
    lo = 1
    hi = n
    res = lo
    while lo <= hi:
        m = (lo + hi) // 2
        if m*m <= n:
            res = max(res, m)
            lo = m + 1
        else:
            hi = m - 1

    return res


ans = 0
s = set()
for x6 in x6list:
    for x4 in x4list:
        x = x6 + x4
        if x > N: break
        if x in s: continue
        s.add(x)

        if x % K == 0:
            z2 = x // K
            z = isq(z2)
            if z*z == z2:
                ans += 1

print(ans)
0