結果
問題 | No.1918 Simple Math ? |
ユーザー |
👑 ![]() |
提出日時 | 2022-04-29 23:27:11 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 451 ms / 2,000 ms |
コード長 | 1,388 bytes |
コンパイル時間 | 781 ms |
コンパイル使用メモリ | 77,336 KB |
最終ジャッジ日時 | 2025-01-29 00:14:26 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 33 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <atcoder/modint>#include <atcoder/fenwicktree>using namespace std;using i32 = int;using u32 = unsigned int;using i64 = long long;using u64 = unsigned long long;#define rep(i,n) for(int i=0; i<(int)(n); i++)using m32 = atcoder::static_modint<1000000007>;i64 isqrt(i64 x){i64 l = 0, r = 2001001001;while(l + 1 < r){i64 m = (l+r) / 2;if(m * m <= x) l = m; else r = m;}return l;}const m32 inv2 = m32(2).inv();const m32 inv6 = m32(6).inv();m32 linear_sum(i64 x){return m32(x) * (m32(x)+1) * inv2;}m32 square_sum(i64 x){return m32(x) * (m32(x)+1) * (m32(x)*2+1) * inv6;}// sum_{i=0}^N ceil(i*i/a)m32 solve2(i64 a, i64 N){//cout << "a = " << a << " , N = " << N << endl;m32 ans = 0;for(i64 b=0; b<a; b++) if(b <= N){i64 n = (N-b) / a;ans += square_sum(n) * a + linear_sum(n) * b * 2 + m32((b*b+a-1)/a)*(n+1);}return ans - N;}m32 solve(){i64 a, N; cin >> a >> N;i64 n = isqrt(a * N);m32 grid = m32(n) * m32(N);return grid - solve2(a, n);}int main(){int T; cin >> T;rep(t,T) cout << solve().val() << '\n';return 0;}struct ios_do_not_sync{ios_do_not_sync(){std::ios::sync_with_stdio(false);std::cin.tie(nullptr);}} ios_do_not_sync_instance;