Question

我有5 X 5矩阵，其中1代表土地，0代表水。需要创建一个将遍历矩阵并返回找到的土地数量的函数。对角1的需要忽略。

我坚持使用我的程序。非常感谢。

var arr = [
    [1, 1, 0, 1, 0],
    [1, 1, 0, 1, 0],
    [0, 1, 0, 0, 0],
    [0, 1, 0, 0, 0],
    [0, 0, 1, 1, 1]
];

我写了一个函数，它使用上述矩阵来查找土地的数量。

function findLand(arr) {

var check = [],
    cntr = 0;

for(let i=0 ;i<arr.length; i++) {
    check = [];
    traverse(i, 0);
}
function traverse(x, y) {
    if(x<0 || y<0 || x > arr.length-1 || y > arr[0].length-1) {
        return;
    }

    if(arr[x][y]!=1 || check.indexOf(x+'_'+y)!=-1) {
        return;
    }
    check.push(x+'_'+y);

    traverse(x, y+1);
    traverse(x, y-1);
    traverse(x-1, y);
    traverse(x+1, y);
}

}

findLand(matrix)

Answer 1

这是一种获取土地块数和座标的方法。您需要遍历整个数组-现在，您只需要遍历第一列，因此您会错过并非始于该边缘的地块。每次迭代都可以创建一个新的空current数组。最后，您将把该数组推入数组“ lands”，这将得到最终结果。我们还将保留一个visited数组，这样您就不必遍历lands矩阵：

var arr = [
    [1, 1, 0, 1, 0],
    [1, 1, 0, 1, 0],
    [0, 1, 0, 0, 0],
    [0, 1, 0, 0, 0],
    [0, 0, 1, 1, 1]
];

function findLand(arr) {
    var lands = [],        // the current group
        visited = new Set  // coords we've seen (set is quicker to lookup)
 
    // iterate the rows AND columns
    for(let i=0; i<arr.length; i++) {
        for(let j=0; j < arr[i].length; j++){
            if (visited.has(i+'_'+j)) continue // don't call function on visited coords
            let land = traverse(i,j)
            if(land) {                // land will be undefined if traverse returns undefined
                lands.push(land);
            }
        } 
    }
    function traverse(x, y, current = []) { // keep current local
        if(x<0 || y<0 || x > arr.length-1 || y > arr[0].length-1) {
            return;
        }
        if(arr[x][y]!=1 || visited.has(x+'_'+y)) {
            return;
        }
        current.push(x+'_'+y);
        visited.add(x+'_'+y)
        traverse(x, y+1, current);
        traverse(x, y-1, current);
        traverse(x-1, y, current);
        traverse(x+1, y, current);
        return current   // should hold one complete land mass
    }
    return lands
}

let lands = findLand(arr)
console.log("lands found: ", lands.length )
console.log("lands: ", lands)

使用递归的JavaScript Flood填充算法

1 个答案: