結果

問題 No.886 Direct
ユーザー ayaoni
提出日時 2020-12-19 09:23:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 713 ms / 4,000 ms
コード長 1,640 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 162,160 KB
最終ジャッジ日時 2024-09-21 10:09:15
合計ジャッジ時間 8,059 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10**7)
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())


def sieve_of_eratosthenes(n):  # n以下の素数の全列挙
    prime_list = []
    is_prime = [1]*(n+1)  # A[i] = iが素数なら1,その他は0
    is_prime[0] = is_prime[1] = 0
    for i in range(2,int(n**.5)+1):
        if is_prime[i]:
            prime_list.append(i)
            for j in range(i**2,n+1,i):
                is_prime[j] = 0
    for i in range(int(n**.5)+1,n+1):
        if is_prime[i] == 1:
            prime_list.append(i)
    return prime_list


H,W = MI()
mod = 10**9+7

if H == 1 or W == 1:
    ans = H*(W-1)+W*(H-1)
    print(ans % mod)
    exit()

f = [H-i for i in range(H+1)]
g = [W-i for i in range(W+1)]

M = max(H,W)
m = min(H,W)
prime_list = sieve_of_eratosthenes(M)

# f,gを約数ゼータ変換する

for p in prime_list:
    for i in range(H//p,0,-1):
        f[i] += f[i*p]
        f[i] %= mod
    for i in range(W//p,0,-1):
        g[i] += g[i*p]
        g[i] %= mod

# 各点積

h = [(f[i]*g[i]) % mod for i in range(m+1)]

# hを約数メビウス変換する

for p in prime_list:
    for i in range(1,m+1):
        if i*p > m:
            break
        h[i] -= h[p*i]
        h[i] %= mod

ans = H*(W-1)+W*(H-1)+2*h[1]
ans %= mod
print(ans)
0