結果

問題 No.2440 Accuracy of Integer Division Approximate Functions
ユーザー vwxyzvwxyz
提出日時 2023-09-13 04:58:35
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 229 ms / 2,000 ms
コード長 725 bytes
コンパイル時間 100 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-06-30 15:10:08
合計ジャッジ時間 5,356 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline=sys.stdin.readline

def floor_sum(n: int, m: int, a: int, b: int) -> int:
    assert 0 <= n
    assert 1 <= m

    ans = 0

    if a >= m:
        ans += (n - 1) * n * (a // m) // 2
        a %= m

    if b >= m:
        ans += n * (b // m)
        b %= m

    y_max = (a * n + b) // m
    x_max = y_max * m - b

    if y_max == 0:
        return ans

    ans += (n - (x_max + a - 1) // a) * y_max
    ans += floor_sum(y_max, a, m, (a - x_max % a) % a)

    return ans

T=int(readline())
for t in range(T):
    N,D,M,S=map(int,readline().split())
    if 2**S-D*M:
        r=min(N+1,(D*2**S)//abs(2**S-D*M)+1)
    else:
        r=N+1
    ans=r-abs(floor_sum(r,D,1,0)-floor_sum(r,2**S,M,0))-1
    print(ans)
0