how do you make a string dictionary function in lu

Question

Is there a way if a string is close to a string in a table it will replace it with the one in the table?

Like a spellcheck function, that searches through a table and if the input is close to one in the table it will fix it , so the one in the table and the string is the same?

Answer 1

You can use this code :) Reference code is from here : https://github.com/badarsh2/Algorithm-Implementations/blob/master/Levenshtein_distance/Lua/Yonaba/levenshtein.lua

local function min(a, b, c)
    return math.min(math.min(a, b), c)
end

local function matrix(row,col)
  local m = {}
  for i = 1,row do m[i] = {}
    for j = 1,col do m[i][j] = 0 end
  end
  return m
end

local function lev(strA,strB)
  local M = matrix(#strA+1,#strB+1)
  local i, j, cost
  local row, col = #M, #M[1]
  for i = 1, row do M[i][1] = i - 1 end
  for j = 1, col do M[1][j] = j - 1 end
  for i = 2, row do
    for j = 2, col do
      if (strA:sub(i - 1, i - 1) == strB:sub(j - 1, j - 1)) then cost = 0
      else cost = 1
      end
    M[i][j] = min(M[i-1][j] + 1,M[i][j - 1] + 1,M[i - 1][j - 1] + cost)
    end
  end
  return M[row][col]
end

local refTable = {"hell", "screen"}

local function getClosestWord(pInput, pTable, threesold)
  cDist = -1
  cWord = ""
  for key, val in pairs(pTable) do
    local levRes = lev(pInput, val)
    if levRes < cDist or cDist == -1 then
      cDist = levRes
      cWord = val
    end
  end
  print(cDist)
  if cDist <= threesold then
    return cWord
  else
    return pInput
  end
end

a = getClosestWord("hello", refTable, 3)
b = getClosestWord("screw", refTable, 3)
print(a, b)

Third parameter is threesold, if min distance is higher than threesold, word is not replaced.