結果
問題 | No.1489 Repeat Cumulative Sum |
ユーザー |
👑 |
提出日時 | 2021-04-30 20:35:43 |
言語 | Lua (LuaJit 2.1.1734355927) |
結果 |
AC
|
実行時間 | 48 ms / 2,000 ms |
コード長 | 1,408 bytes |
コンパイル時間 | 270 ms |
コンパイル使用メモリ | 6,948 KB |
実行使用メモリ | 9,984 KB |
最終ジャッジ日時 | 2024-07-18 18:53:44 |
合計ジャッジ時間 | 3,214 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
local mod = 1000000007local mfl = math.floorlocal function bmul(x, y)local x1, y1 = mfl(x / 31623), mfl(y / 31623)local x0, y0 = x - x1 * 31623, y - y1 * 31623return (x1 * y1 * 14122 + (x1 * y0 + x0 * y1) * 31623 + x0 * y0) % modendlocal function badd(x, y)return (x + y) % modendlocal function bsub(x, y)return x < y and x - y + mod or x - yendlocal function modpow(src, pow)local res = 1while 0 < pow doif pow % 2 == 1 thenres = bmul(res, src)pow = pow - 1endsrc = bmul(src, src)pow = mfl(pow / 2)endreturn resendlocal function modinv(src)return modpow(src, mod - 2)endlocal fact = {1}local invs = {1}local invfact = {1}for i = 2, 200000 dofact[i] = bmul(fact[i - 1], i)invs[i] = bmul(mfl(mod / i), mod - invs[mod % i])invfact[i] = bmul(invfact[i - 1], invs[i])endlocal ffi = require("ffi")local C = ffi.Cffi.cdef[[long long atoll(const char*);]]local function lltonumber(str)return C.atoll(str)endlocal n, m = io.read():match("(%d+) (%d+)")n = tonumber(n)m = lltonumber(m)local a = {}for i = 1, n - 1 doa[i] = io.read("*n")endm = m % 1000000007LLm = tostring(m):gsub("LL", "")m = tonumber(m)local comb = 1local ret = 0for i = 1, n - 1 dolocal ai = a[n - i]comb = bmul(comb, m)comb = bmul(comb, invs[i])ret = badd(ret, bmul(comb, ai))m = (m + 1) % modendprint(ret)