結果
問題 | No.1918 Simple Math ? |
ユーザー |
|
提出日時 | 2022-04-29 23:49:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 609 ms / 2,000 ms |
コード長 | 883 bytes |
コンパイル時間 | 189 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 63,104 KB |
最終ジャッジ日時 | 2024-06-29 05:38:37 |
合計ジャッジ時間 | 9,558 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 33 |
ソースコード
mod = 10**9+7def sqrt(x):ok = 0ng = 10**9+1while ng - ok > 1:d = (ok+ng)>>1if d * d <= x:ok = delse:ng = dreturn okdef f(a, n):ans = 0for k in range(1,sqrt(a*n)+1):ans += n+1-(k*k+a-1)//a#print(n+1-(k*k+a-1)//a)ans %= modreturn ansdef g(a, n):N = sqrt(a*n)ans = (n+1)*Nfor b in range(a):r = (N+b)//al = b//a#print(l, r)ans -= (b*b+a-1)//a * (r - l)ans %= modans -= a * (r*(r+1)*(2*r+1)//6)ans %= modans += a * (l*(l+1)*(2*l+1)//6)ans %= modans += b*(r*(r+1))ans %= modans -= b*(l*(l+1))ans %= modreturn ansfor _ in range(int(input())):a,n = map(int,input().split())#ans1 = f(a, n)ans = g(a, n)print(ans)