結果
問題 | No.1966 Median of Divisors |
ユーザー | terasa |
提出日時 | 2022-06-03 21:53:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 296 ms / 2,000 ms |
コード長 | 970 bytes |
コンパイル時間 | 213 ms |
コンパイル使用メモリ | 82,236 KB |
実行使用メモリ | 77,696 KB |
最終ジャッジ日時 | 2024-09-21 02:40:06 |
合計ジャッジ時間 | 3,488 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 12 |
ソースコード
import sys import pypyjit import itertools import heapq import math from collections import deque, defaultdict import bisect input = sys.stdin.readline sys.setrecursionlimit(10 ** 6) pypyjit.set_param('max_unroll_recursion=-1') def index_lt(a, x): 'return largest index s.t. A[i] < x or -1 if it does not exist' return bisect.bisect_left(a, x) - 1 def index_le(a, x): 'return largest index s.t. A[i] <= x or -1 if it does not exist' return bisect.bisect_right(a, x) - 1 def index_gt(a, x): 'return smallest index s.t. A[i] > x or len(a) if it does not exist' return bisect.bisect_right(a, x) def index_ge(a, x): 'return smallest index s.t. A[i] >= x or len(a) if it does not exist' return bisect.bisect_left(a, x) T = int(input()) mod = 10 ** 9 + 7 for _ in range(T): N, M = map(int, input().split()) s = pow(N, M, mod) t = pow(N, M // 2, mod) print((s * (s + 1) // 2 - t * (t + 1) * (2 * t + 1) // 6) % mod)