結果
問題 | No.1358 [Zelkova 2nd Tune *] 語るなら枚数を... |
ユーザー | persimmon-persimmon |
提出日時 | 2021-02-13 20:28:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,902 ms / 2,000 ms |
コード長 | 1,890 bytes |
コンパイル時間 | 282 ms |
コンパイル使用メモリ | 82,152 KB |
実行使用メモリ | 75,860 KB |
最終ジャッジ日時 | 2024-04-21 05:19:54 |
合計ジャッジ時間 | 11,133 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 55 ms
64,732 KB |
testcase_01 | AC | 37 ms
53,336 KB |
testcase_02 | AC | 39 ms
54,104 KB |
testcase_03 | AC | 37 ms
53,092 KB |
testcase_04 | AC | 38 ms
53,840 KB |
testcase_05 | AC | 38 ms
53,816 KB |
testcase_06 | AC | 78 ms
73,836 KB |
testcase_07 | AC | 76 ms
75,632 KB |
testcase_08 | AC | 73 ms
71,616 KB |
testcase_09 | AC | 73 ms
73,300 KB |
testcase_10 | AC | 68 ms
71,532 KB |
testcase_11 | AC | 1,649 ms
75,860 KB |
testcase_12 | AC | 1,770 ms
75,600 KB |
testcase_13 | AC | 1,902 ms
75,832 KB |
testcase_14 | AC | 1,291 ms
75,792 KB |
testcase_15 | AC | 1,552 ms
75,420 KB |
testcase_16 | AC | 831 ms
75,828 KB |
ソースコード
def main0(n,k,h,y): a,b=0,0 ans=0 while a<=y: b=0 while a+b<=y: yy=y-a-b if yy%h==0: ans+=1 b+=k a+=n return ans mod=10**9+7 def gcd(a,b): while b:a,b=b,a%b return a def xgcd(a, b): x0, y0, x1, y1 = 1, 0, 0, 1 while b != 0: q, a, b = a // b, b, a % b x0, x1 = x1, x0 - q * x1 y0, y1 = y1, y0 - q * y1 return a, x0, y0 memo={} def modinv(a, m): if (a,m) in memo:return memo[(a,m)] g, x, y = xgcd(a, m) if g != 1: raise Exception('modular inverse does not exist') else: memo[(a,m)]=x%m return x % m def main1(n,k,h,y): ary=[n,k,h] ary.sort() a,b,c=ary g=gcd(a,b) na,nb=a//g,b//g now=0 ans=0 while now<=y: yy=y-now now+=c if yy%g!=0:continue yy=yy//g if yy==0: ans+=1 continue if yy<na:continue if yy<nb: if yy%na==0: ans+=1 continue # yy-nb*iがmod naで0になるiの個数 # yy%na=nb*i%na となるiの個数 # 0<=i<=yy//nb # yy%na=nb*i%na となる最小のiを求める。あとは周期naで循環する。 u=yy%na v=nb%na # u=v*i # u*v^(-1)=i if u==0: i=0 elif u==v: i=1 else: i=u*modinv(v,na) i%=na #print((na,nb),yy,i) if yy-nb*i<0:continue w=yy//nb if i<=w: ans+=(w-i)//na+1 ans%=mod #print((u,v,w)) #print((na,nb,yy),i,w,(w-i)//na+1) return ans if __name__=='__main__': t=int(input()) cases=[list(map(int,input().split())) for _ in range(t)] for n,k,h,y in cases: print(main1(n,k,h,y)) if __name__=='__main__1': from random import randint for i in range(100): ary=[randint(1,10) for _ in range(3)] y=randint(10,20) n,k,h=ary ret0=main0(n,k,h,y) ret1=main1(n,k,h,y) if ret0!=ret1: print(1) print(n,k,h,y) break