結果
問題 | No.2891 Mint |
ユーザー |
|
提出日時 | 2024-09-13 23:09:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 113 ms / 2,000 ms |
コード長 | 909 bytes |
コンパイル時間 | 588 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 136,936 KB |
最終ジャッジ日時 | 2024-09-13 23:09:27 |
合計ジャッジ時間 | 5,211 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 54 |
ソースコード
def isqrt(n):if n == 0: return 0x = 1 << (n.bit_length() + 1) // 2y = (x + n // x) // 2while y < x:x = yy = (x + n // x) // 2return ximport sysinput = sys.stdin.readlineMOD = 998244353N, M = map(int, input().split())dp = N*M%MODif N<=isqrt(M) or N<=10**6:for i in range(1, N+1):cnt = M//idp -= i*cnt%MODdp %= MODprint(dp)else:dp = N*M%MODtmp = []for i in range(1, isqrt(M)+1):cnt = M//itmp.append(cnt)dp -= i*cnt%MODdp %= MOD#print(tmp)for i in range(len(tmp)-1):cnt = min(tmp[i], N)-tmp[i+1]st = tmp[i+1]+1if cnt>0:dp -= (i+1)*((st+st+cnt-1)*cnt//2)%MODdp %= MODcnt = tmp[-1]-len(tmp)st = len(tmp)+1if cnt>0:dp -= len(tmp)*((st+st+cnt-1)*cnt//2)%MODdp %= MODprint(dp)