結果

問題 No.2847 Birthday Attack
ユーザー 👑 seekworser
提出日時 2024-07-09 09:08:12
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 304 ms / 3,000 ms
コード長 1,269 bytes
コンパイル時間 7,457 ms
コンパイル使用メモリ 276,328 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-09 09:08:24
合計ジャッジ時間 11,374 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 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;
                if (a*2 < x && b*2 < y) {
                    if (b == 0) ans += (x - 2 * a) * (y - 2 * b) * 2;
                    else ans += (x - 2 * a) * (y - 2 * b) * 4;
                }
                if (a*2 < y && b*2 < x) {
                    if (b == 0) ans += (y - 2 * a) * (x - 2 * b) * 2;
                    else ans += (y - 2 * a) * (x - 2 * b) * 4;
                }
                mul++;
            }
        }
    }
    cout << ans.val() << "\n";
}
0