DFS-油田问题

Posted 2021-11-10 还是不会呀

油田问题

• • 问题描述
  • • Description
    • Input
    • Output
  • JavaScript 解决
  • • 代码
    • 注意事项

涂鸦智能秋招笔试题，很多算法题都在大学时间学到过，但是那时还没有这么完全接触前端，加上也没有去回顾复习，又还给老师了，真是书到用时方恨菜啊！

问题描述

Description

某石油勘探公司正在按计划勘探地下油田资源，工作在一片长方形的地域中。他们首先将该地域划分为许多小正方形区域，然后使用探测设备分别探测每一块小正方形区域内是否有油。若在一块小正方形区域中探测到有油，则标记为’@’，否则标记为’*’。如果两个相邻区域都为 1，那么它们同属于一个石油带，一个石油带可能包含很多小正方形区域，而你的任务是要确定在一片长方形地域中有多少个石油带。 所谓相邻，是指两个小正方形区域上下、左右、左上右下或左下右上同为’@’。

Input

输入数据将包含一些长方形地域数据，每个地域数据的第一行有两个正整数 m 和 n，表示该地域由 mn 个小正方形所组成，如果 m 为 0，表示所有输入到此结束；否则，后面 m（1≤m≤100）行数据，每行有 n（1≤n≤100）个字符，每个字符为’@’或’’，表示有油或无油。每个长方形地域中，’@’的个数不会超过 100。

Output

每个长方形地域，输出油带的个数，每个油带值占独立的一行。油带值不会超过 100。

javascript 解决

代码

function dfs(grid, newGrid, x, y, direction, count) {
  for (let i = 0; i < direction.length; i++) {
    const nextX = x + direction[i][0]; // nextX:下一步要走的x坐标
    const nextY = y + direction[i][1]; // nextY:下一步要走的y坐标
    if (
      nextX >= 0 &&
      nextX < grid.length &&
      nextY >= 0 &&
      nextY < grid[0].length &&
      grid[nextX][nextY] === "@" &&
      !newGrid[nextX][nextY]
    ) {
      // 边界判断+减枝(符合操作再进入下一步遍历)
      newGrid[nextX][nextY] = count;
      dfs(grid, newGrid, nextX, nextY, direction, count);
    }
  }
}

function foo(grid) {
  const row = grid.length; // 获取传入二维数组的行数
  const cloum = grid[0].length; // 获取传入二维数组的列数
  let count = 0; // 个数
  const direction = [
    [0, 1],
    [1, 0],
    [0, -1],
    [-1, 0],
  ];
  const newGrid = JSON.parse(
    JSON.stringify(Array(row).fill(Array(cloum).fill(0)))
  ); // 必须这么做(要么再完成一个深度拷贝函数)
  newGrid[0][0] = 1;
  for (let i = 0; i < row; i++) {
    for (let j = 0; j < cloum; j++) {
      if (grid[i][j] === "@" && !newGrid[i][j]) {
        // 对每个节点进行遍历
        count++;
        newGrid[i][j] = count;
        dfs(grid, newGrid, i, j, direction, count);
      }
    }
  }
  return count;
}
// 测试数据
const grid = [
  ["@", "@", "@", "*", "*"],
  ["@", "@", "@", "*", "@"],
  ["*", "@", "*", "*", "*"],
  ["*", "@", "*", "*", "@"],
  ["*", "@", "@", "*", "*"],
];

console.log(foo(grid));

注意事项

在定义newGrid这个二维数组的时候，之前我是直接:
    const newGrid = Array(row).fill(Array(cloum).fill(0))
但是在测试结果过程中发现：
    newGrid[0][0] = 1;
    console.log(newGrid);
打印结果是：
    [[1,0,0,0,0],[1,0,0,0,0],[1,0,0,0,0],[1,0,0,0,0],[1,0,0,0,0]]

也就是说二维数组newGrid内的数组元素是同一个数组，但我们需要的是，子元素数组每一个都是独立的
那么可以这样解决：
    const newGrid = JSON.parse(JSON.stringify(Array(row).fill(Array(cloum).fill(0))));