結果

問題 No.2847 Birthday Attack
ユーザー 👑 seekworser
提出日時 2024-07-09 23:06:03
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 991 bytes
コンパイル時間 3,288 ms
コンパイル使用メモリ 247,044 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-09 23:06:17
合計ジャッジ時間 6,378 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 2
other WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using ll = long long;
using mint = atcoder::modint;

int main() {
    ll x,y,m; cin >> x >> y >> m;
    mint::set_mod(m);
    auto calc_total = [&] (ll x) -> mint {
        ll xk = x / 2;
        mint xtotal = mint(xk + 1) * xk;
        return mint(xk) * x - xtotal;
    };
    mint ans = calc_total(x) * 2 * y + calc_total(y) * 2 * x;
    ll mx = max(x, y) / 2 + 1;
    for (ll m=1; m<=mx; m++) {
        for (ll n=m+1; n<=mx; n++) {
            if (m % 2 == n % 2 || gcd(n, m) != 1) continue;
            ll ai = n*n - m*m;
            ll bi = 2*n*m;
            if (ai > mx || bi > mx) break;
            ll mul = 1;
            while (mul*max(ai, bi) <= mx) {
                ll a = ai*mul;
                ll b = bi*mul;
                ans += (x - 2 * a) * (y - 2 * b) * 2;
                ans += (y - 2 * a) * (x - 2 * b) * 2;
                mul++;
            }
        }
    }
    cout << ans.val() << "\n";
}
0