結果
問題 | No.1899 L1 Cafe |
ユーザー | taiga0629kyopro |
提出日時 | 2022-04-08 22:56:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 456 ms / 2,000 ms |
コード長 | 700 bytes |
コンパイル時間 | 652 ms |
コンパイル使用メモリ | 82,244 KB |
実行使用メモリ | 77,592 KB |
最終ジャッジ日時 | 2024-11-28 13:53:49 |
合計ジャッジ時間 | 6,195 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 12 |
ソースコード
def floor_sum(n,m,a,b): #Σ i=0→(n-1) [(a*i+b)/m]をO(loga+logm)で求める 1<=n,m a,bはall if not(0<=a<m and 0<=b<m): k=(a-a%m)//m l=(b-b%m)//m return floor_sum(n,m,a%m,b%m)+(l)*n+(k)*n*(n-1)//2 if a==0: return (b//m)*n if n==1: return b//m if a>=m or b>=m: return floor_sum(n,m,a%m,b%m)+(b//m)*n+(a//m)*n*(n-1)//2 y=(a*n+b)//m z=(a*n+b)%m return floor_sum(y,a,m,z) T=int(input()) for _ in range(T): n,a,b=map(int,input().split()) cnt=n//a ansa=(cnt)*(n-a-a*cnt+n)//2 cnt=n//b ansb=(cnt)*(n-b-b*cnt+n)//2 ansab=floor_sum(n//a+1,b,-a,n)-n//b ans=ansa+ansb-ansab print(ans)