結果

問題 No.1918 Simple Math ?
ユーザー googol_S0
提出日時 2022-04-29 21:47:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 665 ms / 2,000 ms
コード長 427 bytes
コンパイル時間 198 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 137,344 KB
最終ジャッジ日時 2024-06-29 03:09:35
合計ジャッジ時間 9,512 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

mod=10**9+7
def solve():
  A,N=map(int,input().split())
  C=[0]*A
  M=A*N
  R=0
  for i in range(31,-1,-1):
    v=R+(1<<i)
    if v*v<=M:
      R=v
  V=0
  for k in range(1,A+1):
    v=(R+A-k)//A
    C[(k*k-1)%A]+=v
  for i in range(A):
    C[i]%=mod
    V=(V+i*C[i])%mod
  W=R*(R+1)*(2*R+1)//6
  W=(W-R)%mod
  W=(W-V)%mod
  W=W*pow(A,mod-2,mod)%mod
  #print(R,N,W)
  print((R*N-W)%mod)

for t in range(int(input())):
  solve()
0