結果

問題 No.1514 Squared Matching
コンテスト
ユーザー 👑 obakyan
提出日時 2021-08-09 23:05:16
言語 Lua
(LuaJit 2.1.1734355927)
結果
MLE  
実行時間 -
コード長 1,664 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 5,376 KB
実行使用メモリ 816,880 KB
最終ジャッジ日時 2024-09-21 22:12:13
合計ジャッジ時間 3,293 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 MLE * 1 -- * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mce, mfl, msq, mmi, mma, mab = math.ceil, math.floor, math.sqrt, math.min, math.max, math.abs

local function comp(a, b)
  return a < b
end

local function lower_bound(ary, x)
  local num = #ary
  if num == 0 then return 1 end
  if not comp(ary[1], x) then return 1 end
  if comp(ary[num], x) then return num + 1 end
  local min, max = 1, num
  while 1 < max - min do
    local mid = mfl((min + max) / 2)
    if comp(ary[mid], x) then
      min = mid
    else
      max = mid
    end
  end
  return max
end

local function getprimes(x)
  local primes = {}
  local allnums = {}
  local alive = {}
  for i = 1, x do allnums[i] = true end
  for i = 1, x do alive[i] = true end
  local withsq = {1}
  for i = 2, x do
    if not alive[i] then
      table.insert(withsq, i)
    end
    if allnums[i] then
      table.insert(primes, i)
      local lim = mfl(x / i)
      for j = 2, lim do
        allnums[j * i] = false
      end
      if 4 < i then
        lim = mfl(x / (i * i))
        for j = 1, lim do
          if 0 < j % 4 and 0 < j % 9 then
            alive[j * i * i] = false
          end
        end
      end
    end
  end
  return primes, withsq
end

local n = io.read("*n")
local primes, withsq = getprimes(n)
-- print(table.concat(withsq, " "))
-- print(#primes, #withsq)
-- print(os.clock())
local t = {}
local baselim = 1
while (baselim + 1) * (baselim + 1) <= n do
  baselim = baselim + 1
end
local ans = baselim * baselim
for base = 1, baselim do
  local cnt = mfl(n / (base * base))
  local lb = lower_bound(withsq, cnt + 1)
  cnt = cnt - mfl(cnt / 4) - mfl(cnt / 9) + mfl(cnt / 36) - (lb - 1)
  ans = ans + cnt * (2 * base - 1)
end
print(ans)
0