結果
問題 | No.665 Bernoulli Bernoulli |
ユーザー | convexineq |
提出日時 | 2021-03-02 20:44:38 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 94 ms / 2,000 ms |
コード長 | 1,002 bytes |
コンパイル時間 | 289 ms |
コンパイル使用メモリ | 87,124 KB |
実行使用メモリ | 79,212 KB |
最終ジャッジ日時 | 2023-07-26 15:02:25 |
合計ジャッジ時間 | 3,013 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 74 ms
78,468 KB |
testcase_01 | AC | 72 ms
78,528 KB |
testcase_02 | AC | 83 ms
78,844 KB |
testcase_03 | AC | 92 ms
79,116 KB |
testcase_04 | AC | 92 ms
79,000 KB |
testcase_05 | AC | 91 ms
79,012 KB |
testcase_06 | AC | 89 ms
78,984 KB |
testcase_07 | AC | 90 ms
78,792 KB |
testcase_08 | AC | 89 ms
78,932 KB |
testcase_09 | AC | 90 ms
79,020 KB |
testcase_10 | AC | 94 ms
79,112 KB |
testcase_11 | AC | 91 ms
79,212 KB |
testcase_12 | AC | 91 ms
79,040 KB |
testcase_13 | AC | 93 ms
79,208 KB |
testcase_14 | AC | 91 ms
79,096 KB |
testcase_15 | AC | 87 ms
78,984 KB |
testcase_16 | AC | 88 ms
79,204 KB |
testcase_17 | AC | 88 ms
78,984 KB |
testcase_18 | AC | 88 ms
79,068 KB |
ソースコード
SIZE=10**5+1; MOD=10**9+7 #998244353 #ここを変更する inv = [0]*SIZE # inv[j] = j^{-1} mod MOD fac = [0]*SIZE # fac[j] = j! mod MOD finv = [0]*SIZE # finv[j] = (j!)^{-1} mod MOD fac[0] = fac[1] = 1 finv[0] = finv[1] = 1 for i in range(2,SIZE): fac[i] = fac[i-1]*i%MOD finv[-1] = pow(fac[-1],MOD-2,MOD) for i in range(SIZE-1,0,-1): finv[i-1] = finv[i]*i%MOD inv[i] = finv[i]*fac[i-1]%MOD def Lagrange_interpolation(a,t): n = len(a)-1 t %= MOD if 0 <= t <= n: return a[t] rprod = [1]*(n+2) r = 1 for i in range(n+2): rprod[n+1-i] = r r = r*(t-n+i)%MOD ans, lprod = 0, 1 for i,ai in enumerate(a): bunsi = lprod*rprod[i+1]%MOD bunbo = finv[i]*finv[n-i]%MOD*(-1 if (n-i)%2 else 1) ans += bunsi*bunbo%MOD*ai%MOD lprod = lprod*(t-i)%MOD return ans%MOD n,k = map(int,input().split()) a = [pow(i,k,MOD) for i in range(2*k+4)] for i in range(1,2*k+4): a[i] += a[i-1] print(Lagrange_interpolation(a,n)%MOD)