結果

問題 No.1358 [Zelkova 2nd Tune *] 語るなら枚数を...
ユーザー rlangevin
提出日時 2023-05-28 16:20:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 750 ms / 2,000 ms
コード長 802 bytes
コンパイル時間 1,083 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 67,968 KB
最終ジャッジ日時 2024-12-27 11:11:44
合計ジャッジ時間 5,181 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
from math import gcd

def extGCD(a, b):
    # ax + by = gcd(a, b)の解を求める
    x, y, u, v = 1, 0, 0, 1
    while b:
        k = a // b
        x -= k * u
        y -= k * v
        x, u = u, x
        y, v = v, y
        a, b = b, a % b
    return x, y


T = int(input())
mod = 10**9 + 7
for _ in range(T):
    N, K, H, Y = map(int, input().split())
    A = [N, K, H]
    A.sort()
    g = gcd(A[0], A[1])
    n, k = A[0]//g, A[1]//g
    p, q = extGCD(n, k)
    ans = 0
    for i in range(Y//A[-1] + 1):
        C = Y - i * A[-1]
        if C < 0:
            break
        if C % g != 0:
            continue
        c = C // g
        ma = c * q // n
        mi = (-c * p + k - 1)//k
        ans += max(0, (ma - mi) + 1)
        ans %= mod
    print(ans)
0