結果

問題 No.886 Direct
ユーザー maspymaspy
提出日時 2020-03-28 10:32:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 222 ms / 4,000 ms
コード長 1,241 bytes
コンパイル時間 390 ms
コンパイル使用メモリ 82,276 KB
実行使用メモリ 78,792 KB
最終ジャッジ日時 2024-06-10 17:10:41
合計ジャッジ時間 4,659 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
56,216 KB
testcase_01 AC 41 ms
55,780 KB
testcase_02 AC 44 ms
56,816 KB
testcase_03 AC 74 ms
74,916 KB
testcase_04 AC 44 ms
55,856 KB
testcase_05 AC 43 ms
55,216 KB
testcase_06 AC 43 ms
55,748 KB
testcase_07 AC 43 ms
55,576 KB
testcase_08 AC 42 ms
56,080 KB
testcase_09 AC 43 ms
55,160 KB
testcase_10 AC 44 ms
56,436 KB
testcase_11 AC 43 ms
57,308 KB
testcase_12 AC 43 ms
55,300 KB
testcase_13 AC 43 ms
56,688 KB
testcase_14 AC 42 ms
55,700 KB
testcase_15 AC 44 ms
55,716 KB
testcase_16 AC 44 ms
56,652 KB
testcase_17 AC 76 ms
76,400 KB
testcase_18 AC 81 ms
76,876 KB
testcase_19 AC 85 ms
77,116 KB
testcase_20 AC 76 ms
76,344 KB
testcase_21 AC 65 ms
73,940 KB
testcase_22 AC 82 ms
76,492 KB
testcase_23 AC 151 ms
77,876 KB
testcase_24 AC 166 ms
77,948 KB
testcase_25 AC 142 ms
77,344 KB
testcase_26 AC 143 ms
77,436 KB
testcase_27 AC 222 ms
78,400 KB
testcase_28 AC 212 ms
78,108 KB
testcase_29 AC 42 ms
55,724 KB
testcase_30 AC 41 ms
56,580 KB
testcase_31 AC 181 ms
78,288 KB
testcase_32 AC 196 ms
78,080 KB
testcase_33 AC 196 ms
78,232 KB
testcase_34 AC 201 ms
78,792 KB
testcase_35 AC 189 ms
78,516 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