結果

問題 No.3156 Count That Day's N
ユーザー norioc
提出日時 2025-06-25 17:34:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 521 ms / 3,000 ms
コード長 768 bytes
コンパイル時間 537 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 233,484 KB
最終ジャッジ日時 2025-06-25 17:34:40
合計ジャッジ時間 13,670 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

from math import isqrt, sqrt

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 = int(sqrt(z2))
            if z*z == z2:
                ans += 1

print(ans)
0