結果

問題 No.3577 フェルマー曲線
コンテスト
ユーザー LyricalMaestro
提出日時 2026-07-05 15:32:59
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 215 ms / 4,000 ms
コード長 794 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 229 ms
コンパイル使用メモリ 96,496 KB
実行使用メモリ 84,184 KB
最終ジャッジ日時 2026-07-05 15:33:04
合計ジャッジ時間 3,658 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

# https://yukicoder.me/problems/no/3577


def main():
    N, B = map(int, input().split())

    b_array = [0] * B
    for b in range(B):
        if b > 0:
            b_array[b] = pow(b, N, B)
        else:
            b_array[b] = 0
    
    b_map = {}
    for b in range(B):
        b_ = b_array[b]
        if b_ not in b_map:
            b_map[b_] = 0
        b_map[b_] += 1    
    b_array = [(b_, count) for b_, count in b_map.items()]

    answer = 0
    for i in range(len(b_array)):
        b1, count1 = b_array[i]
        for j in range(len(b_array)):
            b2, count2 = b_array[j]
            b = (b1 + b2) % B
            if b in b_map:
                v = b_map[b]
                answer += (count1 * count2 * v)
    print(answer)








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