結果

問題 No.1994 Confusing Name
ユーザー 👑 obakyanobakyan
提出日時 2022-07-13 23:09:23
言語 Lua
(LuaJit 2.1.1696795921)
結果
AC  
実行時間 459 ms / 2,000 ms
コード長 1,180 bytes
コンパイル時間 310 ms
コンパイル使用メモリ 5,120 KB
実行使用メモリ 17,888 KB
最終ジャッジ日時 2024-06-25 04:18:18
合計ジャッジ時間 7,457 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 4 ms
5,376 KB
testcase_06 AC 5 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 6 ms
5,376 KB
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 354 ms
15,720 KB
testcase_13 AC 409 ms
15,732 KB
testcase_14 AC 412 ms
15,592 KB
testcase_15 AC 339 ms
16,692 KB
testcase_16 AC 280 ms
14,940 KB
testcase_17 AC 334 ms
16,436 KB
testcase_18 AC 447 ms
17,888 KB
testcase_19 AC 449 ms
15,920 KB
testcase_20 AC 459 ms
15,864 KB
testcase_21 AC 65 ms
5,956 KB
testcase_22 AC 31 ms
5,376 KB
testcase_23 AC 361 ms
16,844 KB
testcase_24 AC 356 ms
15,520 KB
testcase_25 AC 209 ms
9,660 KB
testcase_26 AC 90 ms
6,372 KB
testcase_27 AC 333 ms
15,008 KB
testcase_28 AC 222 ms
10,372 KB
testcase_29 AC 25 ms
5,376 KB
testcase_30 AC 253 ms
13,480 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

local n = io.read("*n", "*l")
local s = {}
local lenmap = {}
local ans = {}
local tmp = {}
for i = 1, n do
  ans[i] = 0
  tmp[i] = ""
  s[i] = io.read()
  local len = #s[i]
  if not lenmap[len] then lenmap[len] = {} end
  table.insert(lenmap[len], i)
end

for len, tbl in pairs(lenmap) do
  local sidx = {}
  for i = 1, #tbl do
    local idx = tbl[i]
    table.insert(sidx, idx)
  end
  for pos = 1, len do
    for i = 1, #tbl do
      local idx = tbl[i]
      if pos == 1 then
        tmp[idx] = s[idx]:sub(2, len)
      elseif pos == len then
        tmp[idx] = s[idx]:sub(1, len - 1)
      else
        tmp[idx] = s[idx]:sub(1, pos - 1) .. s[idx]:sub(pos + 1, len)
      end
    end
    table.sort(sidx, function(a, b) return tmp[a] < tmp[b] end)
    local left = 1
    for i = 1, #tbl - 1 do
      local i1, i2 = sidx[i], sidx[i + 1]
      if tmp[i1] ~= tmp[i2] then
        local z = i - left
        for j = left, i do
          ans[sidx[j]] = ans[sidx[j]] + z
        end
        left = i + 1
      end
    end
    do
      local z = #tbl - left
      for j = left, #tbl do
        ans[sidx[j]] = ans[sidx[j]] + z
      end
    end
  end
end
print(table.concat(ans, "\n"))
0