javascript - 确定数组是等差级数还是等比级数(来自 Coderbyte)

Question

就 coderbyte 而言，这是我的功能代码。但我觉得它不应该这么复杂。我错过了一个简单的技巧吗？

function ArithGeo(arr)
{
    var array_type = -1;
    if (arr.length <= 2) return true;

    var a = arr[1], r = a/arr[0], i;
    for (i = 2; i < arr.length; ++i) {
        if ((a *= r) == arr[i]){
            array_type = "Geometric";
        }
        else{
            array_type = -1;
            break;
        }
    }

    if (array_type == "Geometric")
        return array_type;


    a = arr[1], d = a - arr[0], i;
    for (i = 2; i < arr.length; ++i) {
        if ((a += d) == arr[i]){
            array_type = "Arithmetic";
        }
        else {
            array_type = -1;
            break;
        }
    }
    return array_type;
}

ArithGeo([3,9,15,21,27, 28]);

Answer 1

function ArithGeo(arr) { 

var diff = arr[1] - arr[0];
var ratio = arr[1] / arr[0];

var arith = true;
var geo = true;

for(var i = 0; i < arr.length - 1; i++)
{
    if( arr[i + 1] - arr[i] !== diff )
      arith = false;
    if(arr[i + 1] / ratio !== arr[i])
      geo = false;
}

if(arith === true)
    return "arithmetic";
else if(geo === true)
    return" geometric";
else
    return -1;

}

这也是一个简单的解决方案。我要么在寻找一个几何图案，其中给定元素将被前一个元素整除，要么寻找一个算术图案，其中每个元素增加一个常数。两个变量 diff 和 ratio 包含要在整个数组中搜索的每个模式。

我首先假设 arith 和 geo 为真，如果我找到一个不为真的例子，我将其值设置为假。请注意，您的代码有两个 for 循环，条件完全相同。这是一个很好的迹象，表明您的代码可以压缩到一个循环中。

每次通过循环时，我都会测试是否存在将 arith 或 geo 设置为 false 的条件。最后，在循环退出后，我将确定 arith 或 geo 在整个循环中是否保持为真。如果不是，我返回 - 1 作为来自 Coderbyte 请求的问题。

编辑:关于我的 for 循环条件的快速说明。因为我在每次通过时检查 i + 1 的值，所以我通过将退出条件设置为 arr.length - 1 来确保我不会越界。这样，i + 1 仍然可以到达最后一个元素, 并且一定不会过度。