В Lua, популярном облегченном языке сценариев, работа с вложенными таблицами является обычной задачей. Иногда вам может потребоваться получить индекс или ключ вложенного элемента таблицы. В этой статье мы рассмотрим несколько способов добиться этого, используя простые для понимания примеры кода. Итак, приступим!

Метод 1: итерация с использованием ipairs и рекурсии
Один из подходов — использовать итератор ipairs вместе с рекурсией для обхода вложенной таблицы. Вот пример:

function findIndexRecursive(tbl, target)
  for k, v in ipairs(tbl) do
    if v == target then
      return k
    elseif type(v) == "table" then
      local result = findIndexRecursive(v, target)
      if result then
        return k, result
      end
    end
  end
  return nil
end
-- Usage example
local nestedTable = {
  "apple",
  "banana",
  {"orange", "grape"},
  "kiwi"
}
local index = findIndexRecursive(nestedTable, "grape")
print(index) -- Output: 3, 2

Метод 2: использование ipairs и table.concat
Другой метод предполагает использование итератора ipairs и table.concat для создания строковых представлений вложенной таблицы. Вот пример:

function findIndexConcat(tbl, target)
  local str = table.concat(tbl, ",")
  local index = string.find(str, target)
  if index then
    local _, count = string.gsub(str:sub(1, index), ",", "")
    return count + 1
  end
  return nil
end
-- Usage example
local nestedTable = {
  "apple",
  "banana",
  {"orange", "grape"},
  "kiwi"
}
local index = findIndexConcat(nestedTable, "grape")
print(index) -- Output: 3, 2

Метод 3: использование алгоритма поиска в глубину
Алгоритм поиска в глубину (DFS) может применяться для обхода вложенной таблицы и поиска нужного индекса. Вот пример:

function findIndexDFS(tbl, target)
  local stack = {{tbl, nil}}
  while #stack > 0 do
    local node, parentIndex = unpack(table.remove(stack))
    for i, v in ipairs(node) do
      if v == target then
        return parentIndex, i
      elseif type(v) == "table" then
        table.insert(stack, {v, i})
      end
    end
  end
  return nil
end
-- Usage example
local nestedTable = {
  "apple",
  "banana",
  {"orange", "grape"},
  "kiwi"
}
local index = findIndexDFS(nestedTable, "grape")
print(index) -- Output: 3, 2

В этой статье мы рассмотрели три различных метода получения индекса вложенного элемента таблицы в Lua. Используя такие методы, как рекурсия, манипуляции со строками и алгоритмы поиска в глубину, вы можете эффективно перемещаться и получать нужный индекс. Не забудьте выбрать метод, который лучше всего подходит для вашего конкретного случая использования.