#746236 (Lua) No.1876 Xor of Sum

• 問題: No.1876 Xor of Sum
• ユーザー: 👑 obakyan
• 提出日時: 2022-03-16 09:55:03
• 言語: Lua (LuaJit 2.1.1696795921)
• 結果: AC
• 実行時間: 1,235 ms / 2,000 ms
• コード長: 1,684 bytes
• コンパイル時間: 188 ms
• コンパイル使用メモリ: 6,944 KB
• 実行使用メモリ: 6,784 KB
• 最終ジャッジ日時: 2024-09-23 06:49:34
• 合計ジャッジ時間: 15,300 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	120 ms; 5,376 KB
testcase_04	AC	344 ms; 5,376 KB
testcase_05	AC	9 ms; 5,376 KB
testcase_06	AC	629 ms; 5,376 KB
testcase_07	AC	13 ms; 5,376 KB
testcase_08	AC	10 ms; 5,376 KB
testcase_09	AC	281 ms; 5,376 KB
testcase_10	AC	85 ms; 5,376 KB
testcase_11	AC	448 ms; 5,376 KB
testcase_12	AC	119 ms; 5,376 KB
testcase_13	AC	264 ms; 5,376 KB
testcase_14	AC	185 ms; 5,376 KB
testcase_15	AC	219 ms; 5,376 KB
testcase_16	AC	8 ms; 5,376 KB
testcase_17	AC	516 ms; 5,376 KB
testcase_18	AC	642 ms; 5,376 KB
testcase_19	AC	625 ms; 5,376 KB
testcase_20	AC	625 ms; 5,376 KB
testcase_21	AC	635 ms; 5,376 KB
testcase_22	AC	649 ms; 5,376 KB
testcase_23	AC	1,235 ms; 5,376 KB
testcase_24	AC	1,227 ms; 5,376 KB
testcase_25	AC	1,232 ms; 5,376 KB
testcase_26	AC	1,227 ms; 5,376 KB
testcase_27	AC	1,227 ms; 5,376 KB
testcase_28	AC	1,200 ms; 6,784 KB
testcase_29	AC	3 ms; 5,376 KB

ソースコード

local mfl, mce = math.floor, math.ceil
local bls, brs = bit.lshift, bit.rshift
local bor, band = bit.bor, bit.band
local bxor = bit.bxor
local TFDP = {}

TFDP.initialize = function(self, lim)
  self.lim = lim
  self.sz = 30
  self.way = true
  self.bcnt = mce((lim + 1) / self.sz)
  self.b1, self.b2 = {1}, {}
  for i = 2, self.bcnt do
    self.b1[i] = 0
    self.b2[i] = 0
  end
  self.curmax = 0
end

TFDP.add = function(self, v, cursum)
  local src = self.way and self.b1 or self.b2
  local dst = self.way and self.b2 or self.b1
  self.way = not self.way
  local bcnt = self.bcnt
  local bc2 = mce((cursum + 1) / self.sz)
  for i = 1, bc2 do
    dst[i] = src[i]
  end
  local sz = self.sz
  local jump, rem = mfl(v / sz), v % sz
  local denom, mul = bls(1, sz - rem), bls(1, rem)
  for i = 1, bcnt - jump do
    if bc2 < i then break end
    local v1 = src[i] % denom
    dst[i + jump] = bxor(dst[i + jump], v1 * mul)
  end
  for i = 1, bcnt - jump - 1 do
    if bc2 < i then break end
    local v2 = brs(src[i], sz - rem)
    dst[i + jump + 1] = bxor(dst[i + jump + 1], v2)
  end
end

TFDP.query = function(self, x)
  local group = mfl(x / self.sz) + 1
  local idx = x - (group - 1) * self.sz
  local tbl = self.way and self.b1 or self.b2
  return band(tbl[group], bls(1, idx)) ~= 0
end

local tf = TFDP
local n = io.read("*n")
local asum = 0
local a = {}
for i = 1, n do
  a[i] = io.read("*n")
  asum = asum + a[i]
end
tf:initialize(asum)
local cursum = 0
for i = 1, n do
  local ai = a[i]
  tf:add(ai, cursum)
  cursum = cursum + ai
end
-- print(os.clock())
local ret = 0
for i = 1, asum do
  if tf:query(i) then
    ret = bxor(ret, i)
  end
end
print(ret)
-- print(os.clock())