結果

問題 No.2131 Concon Substrings (COuNt Version)
ユーザー 👑 obakyan
提出日時 2022-11-26 21:38:47
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 22 ms / 2,000 ms
コード長 848 bytes
コンパイル時間 371 ms
コンパイル使用メモリ 6,816 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-03 04:39:09
合計ジャッジ時間 1,183 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mod = 998244353
local mfl = math.floor
local function bmul(x, y)
  local x0, y0 = x % 31596, y % 31596
  local x1, y1 = mfl(x / 31596), mfl(y / 31596)
  return (x1 * y1 * 62863 + (x1 * y0 + x0 * y1) * 31596 + x0 * y0) % mod
end
local function badd(x, y)
  return (x + y) % mod
end

local n = io.read("*n")
local a, b, c = {}, {}, {}
local lim = 1002
for i = 1, lim do
  a[i], b[i], c[i] = 0, 0, 0
end
local nothing = 1
for i = 1, n do
  for j = lim, 1, -1 do
    c[j] = badd(c[j] * 25, b[j])
    b[j] = badd(b[j] * 25, a[j])
    if 1 < j then
      a[j] = badd(a[j] * 25, c[j - 1])
    else
      a[j] = badd(a[j] * 25, nothing)
    end
  end
  nothing = bmul(nothing, 25)
end
local ret = 0
for i = 1, lim do
  local x = badd(a[i], b[i])
  x = bmul(x, i - 1)
  local y = bmul(c[i], i)
  ret = badd(ret, x)
  ret = badd(ret, y)
end
print(ret)
0