結果

問題 No.822 Bitwise AND
コンテスト
ユーザー 👑 obakyan
提出日時 2019-04-26 22:40:37
言語 Lua
(LuaJit 2.1.1734355927)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,080 bytes
コンパイル時間 37 ms
コンパイル使用メモリ 6,820 KB
実行使用メモリ 16,696 KB
最終ジャッジ日時 2024-06-25 18:14:53
合計ジャッジ時間 6,464 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other TLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

local n, k = io.read("*n", "*n")
-- 2^a + n - (2^a - 1) <= k
if(n < k) then print("INF")
else
  local len = 0
  local lastzero = -1
  local zeropos = {}
  local tmpn = n
  while(0 < tmpn) do
    len = len + 1
    local s = tmpn % 2
    if(s == 0) then
      lastzero = len
      table.insert(zeropos, len)
    end
    tmpn = math.floor(tmpn / 2)
  end
  if(lastzero == -1) then
    print(1)
  else
    local allcnt = 0
    local a_base = 0
    local arb_count = #zeropos
    local iter = 3^arb_count - 1
    for i = 0, iter do
      local i_tmp = i
      local a_cand = a_base
      local b_cand = 0
      local mult = 1
      for k_tmp = 1, arb_count do
        local s = i_tmp % 3
        if(s == 0) then a_cand = a_cand + 2^(zeropos[mult] - 1)
        elseif(s == 1) then b_cand = b_cand + 2^(zeropos[mult] - 1)
        end
        mult = mult + 1
        i_tmp = math.floor(i_tmp / 3)
      end
      local flg = (b_cand <= a_cand) and (a_cand - b_cand <= k)
      --print(i, a_cand, b_cand, flg)
      if(flg) then allcnt = allcnt + 1 end
    end
    print(allcnt)
  end
end
0