結果
問題 | No.1235 ζ関数 |
ユーザー | kit84 |
提出日時 | 2020-09-22 20:39:23 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 246 ms / 2,000 ms |
コード長 | 1,473 bytes |
コンパイル時間 | 183 ms |
コンパイル使用メモリ | 13,056 KB |
実行使用メモリ | 34,724 KB |
最終ジャッジ日時 | 2024-06-26 11:13:51 |
合計ジャッジ時間 | 6,147 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 234 ms
34,476 KB |
testcase_01 | AC | 243 ms
34,468 KB |
testcase_02 | AC | 235 ms
34,604 KB |
testcase_03 | AC | 235 ms
34,724 KB |
testcase_04 | AC | 238 ms
34,472 KB |
testcase_05 | AC | 241 ms
34,724 KB |
testcase_06 | AC | 242 ms
34,724 KB |
testcase_07 | AC | 236 ms
34,604 KB |
testcase_08 | AC | 244 ms
34,600 KB |
testcase_09 | AC | 239 ms
34,472 KB |
testcase_10 | AC | 246 ms
34,600 KB |
testcase_11 | AC | 227 ms
34,596 KB |
testcase_12 | AC | 234 ms
34,596 KB |
testcase_13 | AC | 238 ms
34,596 KB |
testcase_14 | AC | 237 ms
34,604 KB |
testcase_15 | AC | 234 ms
34,476 KB |
testcase_16 | AC | 239 ms
34,600 KB |
testcase_17 | AC | 239 ms
34,472 KB |
testcase_18 | AC | 231 ms
34,596 KB |
testcase_19 | AC | 243 ms
34,468 KB |
testcase_20 | AC | 239 ms
34,472 KB |
ソースコード
import sys #import collections as col input=sys.stdin.readline enum=enumerate inf=1001001001 eps=1e-7 upb=1e+4 lmax=300 mod = 10**9+7 NN = 2 * 10**5 g1 = [1, 1] # factorial g2 = [1, 1] # factorial^(-1) inverse = [0, 1] # tmp def linput(ty=int, cvt=list): return cvt(map(ty,input().split())) def vinput(rep=1, ty=int, cvt=list): return cvt(ty(input().rstrip()) for _ in "*"*rep) def modinv(x, mod=mod): return pow(x, mod-2, mod) def ratioal(x, y, mod=mod): ### = x/y (mod m) re = x * modinv(y) return re % mod def facto(x, y, mod): ### y*(y+1)*...*(x-1)*x ### = x!/(y-1)! re = 1 for i in range(y, x+1): re *= i re %= mod #print(re, file=sys.stderr) return re def comb(n, r, mod=mod): if ( r<0 or r>n ): return 0 #r = min(r, n-r) #return facto(n, n-r+1, mod) * g2[r] % mod return g1[n] * g2[r] * g2[n-r] % mod def init_fac(): for i in range( 2, NN + 1 ): g1.append( ( g1[-1] * i ) % mod ) inverse.append( ( -inverse[mod % i] * (mod//i) ) % mod ) g2.append( (g2[-1] * inverse[-1]) % mod ) init_fac() def zeta(s): out = 0. pp = float(inf) for m in range(1,lmax+1): ins = 0. for j in range(1,m+1): c1 = (-1)**(j-1) c2 = comb(m-1, j-1) c3 = j**(-s) ins += c1*c2*c3 ins *= 2**(-m) / (1 - 2**(1-s)) out += ins if abs(pp - ins) < eps: break if abs(out) > upb: break pp = ins return out def main(): N, = linput() res = zeta(N) print(res) if __name__=="__main__": main()