結果

問題 No.3156 Count That Day's N
コンテスト
ユーザー cologne
提出日時 2025-05-27 15:19:48
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 503 bytes
コンパイル時間 665 ms
コンパイル使用メモリ 82,608 KB
実行使用メモリ 76,644 KB
最終ジャッジ日時 2025-05-27 15:19:54
合計ジャッジ時間 4,898 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 


import math


def main():
    k, n = map(int, input().split())
    s, q = [], []
    for i in range(1, n+1):
        if i**6 <= n:
            s.append(i**6)
        if i**4 <= n:
            q.append(i**4)
        else:
            break

    qq = set()
    for x in s:
        for y in q:
            if (x+y) % k != 0 or x+y > n:
                continue
            z = (x+y)//k
            if math.isqrt(z)**2 == z:
                qq.add(z)
    print(qq)


if __name__ == '__main__':
    main()
0