結果

問題 No.2624 Prediction by Average
コンテスト
ユーザー 寝癖
提出日時 2024-02-09 23:19:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 809 bytes
コンパイル時間 434 ms
コンパイル使用メモリ 82,144 KB
実行使用メモリ 77,588 KB
最終ジャッジ日時 2024-09-28 16:31:27
合計ジャッジ時間 1,661 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 2 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

def floor_sum(n,m,a,b):
    ans=0
    if a<0:
        a2=a%m
        return floor_sum(n,m,a2,b)-n*(n-1)*((a2-a)//m)//2
    if b<0:
        b2=b%m
        return floor_sum(n,m,a,b2)-n*((b2-b)//m)
    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(input())
for _ in range(T):
    N, S = input().split()
    N = int(N)
    S1000= int(S.replace('.', ''))
    if S1000 == 100000:
        print(N)
        continue

    A = floor_sum(N+1, 1000, S1000+1, -1)-floor_sum(1, 1000, S1000+1, -1)
    B = floor_sum(N+1, 1000, S1000, -1)-floor_sum(1, 1000, S1000, -1)

    print(A-B)
0