結果

問題 No.1918 Simple Math ?
ユーザー Vladimir Novikov
提出日時 2025-10-02 07:14:15
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,705 bytes
コンパイル時間 2,146 ms
コンパイル使用メモリ 196,276 KB
実行使用メモリ 7,720 KB
最終ジャッジ日時 2025-10-02 07:14:22
合計ジャッジ時間 6,388 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 2 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define all(a) begin(a), end(a)
#define rall(a) rbegin(a), rend(a)
#define len(a) (int)((a).size())

const ll INF = 1'000'000'000'000'000'000, MOD = 1'000'000'007;

ll bpow(ll a, ll p) {
    if (p == 0) return 1ll;
    if (p % 2) return (bpow(a, p - 1) * a) % MOD;
    ll b = bpow(a, p / 2);
    return b * b % MOD;
}

ll root(ll a) {
    ll v = sqrt(a);
    while (v * v > a) --v;
    while ((v + 1) * (v + 1) <= a) ++v;
    return v;
}

ll mod_sum(ll n, ll add, ll dv) {
    if (n == 1) {
        return (1 + add) / dv;
    }
    
    ll ans = 0;
    for (ll x = 0; x < dv; ++x) {
        ll good_vals = (n - x);
        if (x == 0) good_vals -= (dv - 1);
        good_vals += (dv - 1);
        good_vals /= dv;
        ll dvmod = (x * x + add) % dv;
        ans += good_vals * dvmod;
        ans %= MOD;
    }

    return ans;
}

ll sq(ll n) {
    n %= MOD;
    return n % MOD * (n + 1) % MOD * (2 * n + 1) % MOD * bpow(6, MOD - 2) % MOD;
}

ll sq_sum(ll n, ll add, ll dv) {
    ll sm = (sq(n) + add * n) % MOD;
    ll md = mod_sum(n, add, dv);
    return (sm - md + MOD) % MOD  * bpow(dv, MOD - 2) % MOD;
}

// ll sq_sum(ll n, ll add, ll dv) {
//     ll calc = 0;
//     for (ll v = 1; v <= n; ++v) {
//         calc += (v * v + add) / dv;
//         calc %= MOD;
//     }
//     return calc;
// }

ll solve(ll a, ll n) {
    return ((n + 1) % MOD * (root(a * n)) % MOD - sq_sum(root(a * n), a - 1, a) + MOD) % MOD;
}

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

    int t;
    cin >> t;
    while (t--) {
        ll a, n;
        cin >> a >> n;
        cout << solve(a, n) << endl;
    }

    return 0;
}
0