結果

問題 No.3577 フェルマー曲線
コンテスト
ユーザー Mage
提出日時 2026-07-03 23:24:32
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 304 ms / 4,000 ms
コード長 2,045 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,255 ms
コンパイル使用メモリ 333,496 KB
実行使用メモリ 6,528 KB
最終ジャッジ日時 2026-07-03 23:25:42
合計ジャッジ時間 5,965 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;

// ================== MACROS ==================
#define ll long long
#define ull unsigned long long
#define ld long double
#define endl '\n'
#define str string
#define ld long double
#define vld vector<ld>
#define vi vector<int>
#define vb vector<bool>
#define vstr vector<str>
#define umapii unordered_map<int, int>
#define vvi vector<vi>
#define vll vector<ll>
#define pii pair<int, int>
#define pll pair<ll, ll>
#define vpii vector<pair<int, int>>
#define i128 __int128_t
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define pb push_back
#define ff first
#define ss second

// ================== CONSTANTS ==================
const ll INF = 1e18;
// const int MOD = 1e9 + 7;  // leetcode
const int MOD = 998244353;  // atcoder, codeforces
// const int MOD = 1001113;  // kattis
const int N = 2e5 + 5;
string v = "aeiou";
string V = "AEIOU";

// ================== MATH ==================
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }

ll mod_add(ll a, ll b) { return (a % MOD + b % MOD) % MOD; }
ll mod_mul(ll a, ll b) { return (a % MOD * b % MOD) % MOD; }
ll mod_pow(ll a, ll b) {
  ll res = 1;
  while (b) {
    if (b & 1) res = mod_mul(res, a);
    a = mod_mul(a, a);
    b >>= 1;
  }
  return res;
}
ll mod_pow(ll a, ll b, ll mod) {
  ll res = 1 % mod;
  a %= mod;
  while (b) {
    if (b & 1) res = (__int128)res * a % mod;
    a = (__int128)a * a % mod;
    b >>= 1;
  }
  return res;
}

// ================== SOLUTION ==================
void solve() {
  ll n, b;
  cin >> n >> b;

  vector<ll> freq(b, 0);

  for (ll x = 0; x < b; x++) {
    ll v = mod_pow(x, n, b);
    freq[v]++;
  }

  ll ans = 0;
  for (ll a = 0; a < b; a++) {
    for (ll c = 0; c < b; c++) {
      ans += freq[a] * freq[c] * freq[(a + c) % b];
    }
  }

  cout << ans << '\n';
}

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);

  int t = 1;
  // cin >> t;
  while (t--) {
    solve();
  }

  return 0;
}
0