結果

問題 No.2280 FizzBuzz Difference
ユーザー sotanishysotanishy
提出日時 2023-04-21 23:35:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,290 bytes
コンパイル時間 378 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 78,780 KB
最終ジャッジ日時 2024-11-06 16:54:33
合計ジャッジ時間 1,839 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,864 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 AC 90 ms
76,472 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

def extgcd(a, b):
    s, sx, sy, t, tx, ty = a, 1, 0, b, 0, 1
    while t:
        q = s // t
        s -= t * q
        s, t = t, s
        sx -= tx * q
        sx, tx = tx, sx
        sy -= ty * q
        sy, ty = ty, sy
    return sx, sy


def calc(A, B, K, M):
    # find the number of (x,y)
    # such that Ax+K=By
    # and 1 <= x <= M/A
    # and 1 <= y <= M/B

    g = gcd(A, B)
    if K % g != 0:
        return 0
    A //= g
    B //= g
    K //= g

    x0, y0 = extgcd(-A, B)
    x0 *= K
    y0 *= K

    # x = x0 + Bt
    # y = y0 + At

    tmin = max((1-x0+B-1)//B, (1-y0+A-1)//A)
    tmax = min((M//A-x0)//B, (M//B-y0)//A)
    return max(0, tmax-tmin+1)


T = int(input())
for _ in range(T):
    M, A, B, K = map(int, input().split())

    if K > A:
        print(0)
        continue

    if B % A == 0:
        if K == A:
            print(M//A-1)
        else:
            print(0)
        continue

    if A == K:
        C = M//A-1
        lcm = A*B//gcd(A, B)
        cntBnotA = M//B - M//lcm
        cnt_all = M//A + cntBnotA
        ans = cnt_all - 1 - 2*cntBnotA
        if M % B < M % A:
            ans += 1
        print(ans)
        continue

    ans = calc(A, B, K, M) + calc(B, A, K, M)
    print(ans)
0