Codility: 在Lua中通过汽车

我目前正在练习编程问题，出于兴趣，我正在尝试在Lua中尝试一些Codility练习。我陷在了“通过汽车”问题上。

问题：

给定一个由N个整数组成的非空的零索引数组A。数组A的连续元素表示道路上的连续汽车。

数组A仅包含0或1：

0表示向东行驶的汽车，
1表示向西行驶的汽车。
目标是计算通过车辆的数量。当P向东行驶而Q向西方行驶时，我们说成对的车辆（P，Q）（0≤P＜Q ＜N）正在通过。

例如，考虑这样的数组A：

  A [0] = 0
  A [1] = 1
  A [2] = 0
  A [3] = 1
  A [4] = 1
我们有五对通过汽车：（0,1），（0,3），（0,4），（2,3），（2,4）。

编写一个函数：

function solution（A）

给定一个由N个整数组成的非空的零索引数组A，返回通过汽车的对数。

如果通过汽车的对数超过1,000,000,000，则应该返回-1.

例如，给定：

  A [0] = 0
  A [1] = 1
  A [2] = 0
  A [3] = 1
  A [4] = 1
函数应返回5，如上所述。

假设：

N是[1..100,000]范围内的整数;
数组A的每个元素都是可以具有以下值之一的整数：0,1。
复杂度：

预期最坏情况下的时间复杂度为O（N）;
预期的最坏情况下的空间复杂度是O（1），超出输入存储（不计算输入参数所需的存储）。
输入数组的元素可以被修改。

我在Lua中的尝试不断失败，但我似乎找不到问题所在。

local function solution(A)
    local zeroes = 0
    local pairs = 0
    for i = 1, #A do
        if A[i] == 0 then
           zeroes = zeroes + 1
        else
            pairs = pairs + zeroes
            if pairs > 1e9 then
                return -1
            end
        end
    end
    return pairs
end

就时间和空间复杂性约束而言，我认为它应该通过，所以我似乎找不到问题所在。我做错了什么？欢迎任何建议或有助于使我的代码更有效的提示。 FYI：我的结果一直是2，而所需的示例结果是5。