結果

問題 No.886 Direct
ユーザー 👑 maspymaspy
提出日時 2020-03-28 10:32:19
言語 PyPy3
(7.3.1)
結果
AC  
実行時間 278 ms / 4,000 ms
コード長 1,241 Byte
コンパイル時間 1,397 ms
使用メモリ 85,052 KB
最終ジャッジ日時 2020-09-12 18:06:20
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 83 ms
80,564 KB
testcase_01 AC 84 ms
80,532 KB
testcase_02 AC 89 ms
80,152 KB
testcase_03 AC 116 ms
81,880 KB
testcase_04 AC 83 ms
80,576 KB
testcase_05 AC 83 ms
80,636 KB
testcase_06 AC 82 ms
80,556 KB
testcase_07 AC 84 ms
80,560 KB
testcase_08 AC 83 ms
80,180 KB
testcase_09 AC 81 ms
80,472 KB
testcase_10 AC 83 ms
80,452 KB
testcase_11 AC 83 ms
80,396 KB
testcase_12 AC 84 ms
80,560 KB
testcase_13 AC 87 ms
80,164 KB
testcase_14 AC 82 ms
80,384 KB
testcase_15 AC 82 ms
80,560 KB
testcase_16 AC 81 ms
80,644 KB
testcase_17 AC 112 ms
81,852 KB
testcase_18 AC 119 ms
82,152 KB
testcase_19 AC 122 ms
82,692 KB
testcase_20 AC 112 ms
82,116 KB
testcase_21 AC 105 ms
81,540 KB
testcase_22 AC 119 ms
81,704 KB
testcase_23 AC 199 ms
83,840 KB
testcase_24 AC 232 ms
83,544 KB
testcase_25 AC 203 ms
84,464 KB
testcase_26 AC 203 ms
83,808 KB
testcase_27 AC 278 ms
84,292 KB
testcase_28 AC 267 ms
85,052 KB
testcase_29 AC 84 ms
80,576 KB
testcase_30 AC 83 ms
80,564 KB
testcase_31 AC 241 ms
84,532 KB
testcase_32 AC 252 ms
84,216 KB
testcase_33 AC 253 ms
84,428 KB
testcase_34 AC 260 ms
84,724 KB
testcase_35 AC 245 ms
84,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
from functools import lru_cache

H, W = map(int, readline().split())
if H < W:
    H, W = W, H


def S1(n):
    return n * (n + 1) // 2


def S2(n):
    return n * (n + 1) * (2 * n + 1) // 6


@lru_cache(None)
def f(H, W):
    """ 1 <= x <= H, 1 <= y <= W, gcd(x,y) = 1 に対して、sum 1, sum x, sum y, sum xyをかえす"""
    SH = H * (H + 1) // 2
    SW = W * (W + 1) // 2
    sum_1 = H * W
    sum_x = SH * W
    sum_y = SW * H
    sum_xy = SH * SW
    d = 2
    while True:
        Hd = H // d
        Wd = W // d
        if not Wd:
            break
        next_d = min(H // Hd, W // Wd) + 1
        s1, sx, sy, sxy = f(Hd, Wd)
        sum_1 -= s1 * (next_d - d)
        sum_x -= sx * (S1(next_d - 1) - S1(d - 1))
        sum_y -= sy * (S1(next_d - 1) - S1(d - 1))
        sum_xy -= sxy * (S2(next_d - 1) - S2(d - 1))
        d = next_d
    return sum_1, sum_x, sum_y, sum_xy


sum_1, sum_x, sum_y, sum_xy = f(H, W)
answer = sum_1 * H * W - W * sum_x - H * sum_y + sum_xy
answer *= 2
# まよこ、まうえ
answer += (W - 1) * H
answer += (H - 1) * W

MOD = 10 ** 9 + 7
print(answer % MOD)

0