結果

問題 No.2847 Birthday Attack
ユーザー 👑 AngrySadEightAngrySadEight
提出日時 2024-07-04 23:30:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,928 bytes
コンパイル時間 836 ms
コンパイル使用メモリ 84,660 KB
実行使用メモリ 54,532 KB
最終ジャッジ日時 2024-07-05 01:29:46
合計ジャッジ時間 10,911 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 334 ms
54,384 KB
testcase_01 AC 336 ms
53,040 KB
testcase_02 AC 335 ms
53,100 KB
testcase_03 AC 349 ms
54,048 KB
testcase_04 AC 337 ms
52,976 KB
testcase_05 AC 773 ms
52,776 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <utility>
#include <vector>
using namespace std;
using ll = long long;
using pll = pair<ll, ll>;

ll mod = 998244535;

ll my_gcd(ll a, ll b) {
    ll ret = 0;
    if (a < b) {
        swap(a, b);
    }
    if (b == 0) {
        ret = a;
    } else {
        ret = my_gcd(b, a % b);
    }
    return ret;
}

int main() {
    ll X, Y;
    cin >> X >> Y;
    ll ans = 0;
    for (ll i = 1; i <= X; i++) {
        ans += 2 * max(X - 2 * i, 0LL) * Y;
        ans %= mod;
    }
    for (ll i = 1; i <= Y; i++) {
        ans += 2 * max(Y - 2 * i, 0LL) * X;
        ans %= mod;
    }
    // ピタゴラス数を列挙する
    vector<pair<ll, pll>> p(0);
    for (ll m = 1; m * m <= 4000000; m++) {
        for (ll n = m + 1; n * n <= 4000000; n++) {
            if (m * m + n * n > 8000000) {
                continue;
            }
            if (m % 2 == n % 2) {
                continue;
            }
            ll g = my_gcd(n, m);
            if (g != 1) {
                continue;
            }
            ll min_p = 2 * m * n;
            ll max_p = n * n - m * m;
            pll pp1 = pll(min_p, max_p);
            pll pp2 = pll(max_p, min_p);
            pair<ll, pll> ppp1 = pair<ll, pll>(m * m + n * n, pp1);
            pair<ll, pll> ppp2 = pair<ll, pll>(m * m + n * n, pp2);
            p.push_back(ppp1);
            p.push_back(ppp2);
        }
    }
    sort(p.begin(), p.end());
    // 列挙されたピタゴラス数の全てに対して、斜めのぶんの数え上げを行う
    ll len_p = p.size();
    for (ll i = 0; i < len_p; i++) {
        ll x1 = p[i].second.first;
        ll y1 = p[i].second.second;
        for (ll j = 1; j <= min(X / y1, Y / x1); j++) {
            ll x = x1 * j;
            ll y = y1 * j;
            ans += 4 * max(Y - 2 * x, 0LL) * max(X - 2 * y, 0LL);
            ans %= mod;
        }
    }
    cout << ans << endl;
}
0