結果

問題 No.1918 Simple Math ?
コンテスト
ユーザー karinohito
提出日時 2022-03-26 11:02:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 166 ms / 2,000 ms
コード長 732 bytes
コンパイル時間 4,170 ms
コンパイル使用メモリ 251,048 KB
最終ジャッジ日時 2025-01-28 12:54:02
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include<atcoder/all>
using namespace atcoder;
using namespace std;
using ll = long long;
ll A, N,T;
using mint = modint1000000007;
mint B[1000006];
ll sqrtz(ll N) {
    ll L = 0, R = 2e9;
    while (abs(R - L) > 1) {
        ll mid = (R + L) / 2;
        if (mid * mid <= N)L = mid;
        else R = mid;
    }
    return L;
}
mint f(ll m) {
    ll r = sqrtz(m * A);
    mint R = r;
    mint res = r * mint(m);
    mint D = R * (R - 1) * (2 * R + 5) / 6 - mint(r / A) * B[A] - B[r % A];
    return res - D / A;
}

int main() {
    cin >> T;
    while(T){
        cin >> A >> N;
        T--;
        for (ll i = 1; i <= A; i++)B[i] = B[i - 1] + ((i * i - 1) % A);
        cout << f(N).val() << endl;
    }
}
0