結果

問題 No.1918 Simple Math ?
コンテスト
ユーザー Vladimir Novikov
提出日時 2025-10-02 05:49:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 849 bytes
コンパイル時間 5,811 ms
コンパイル使用メモリ 195,516 KB
実行使用メモリ 15,952 KB
最終ジャッジ日時 2025-10-02 05:49:28
合計ジャッジ時間 12,406 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 13 WA * 2 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

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 root(ll a) {
    ll v = sqrt(a);
    while (v * v > a) --v;
    while ((v + 1) * (v + 1) <= a) ++v;
    return v;
}


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) * (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