leetcode icon indicating copy to clipboard operation
leetcode copied to clipboard

1878. 矩阵中最大的三个菱形和

Open buuing opened this issue 4 years ago • 0 comments

给你一个 m x n 的整数矩阵 grid 。

菱形和 指的是 grid 中一个正菱形 边界 上的元素之和。本题中的菱形必须为正方形旋转45度,且四个角都在一个格子当中。下图是四个可行的菱形,每个菱形和应该包含的格子都用了相应颜色标注在图中。

image

注意,菱形可以是一个面积为 0 的区域,如上图中右下角的紫色菱形所示。

请你按照 降序 返回 grid 中三个最大的 互不相同的菱形和 。如果不同的和少于三个,则将它们全部返回。

示例 1:

image

输入:grid = [[3,4,5,1,3],[3,3,4,2,3],[20,30,200,40,10],[1,5,5,4,1],[4,3,2,2,5]]
输出:[228,216,211]
解释:最大的三个菱形和如上图所示。
- 蓝色:20 + 3 + 200 + 5 = 228
- 红色:200 + 2 + 10 + 4 = 216
- 绿色:5 + 200 + 4 + 2 = 211

示例 2:

image

输入:grid = [[1,2,3],[4,5,6],[7,8,9]]
输出:[20,9,8]
解释:最大的三个菱形和如上图所示。
- 蓝色:4 + 2 + 6 + 8 = 20
- 红色:9 (右下角红色的面积为 0 的菱形)
- 绿色:8 (下方中央面积为 0 的菱形)

示例 3:

输入:grid = [[7,7,7]]
输出:[7]
解释:所有三个可能的菱形和都相同,所以返回 [7] 。

提示:

  • m == grid.length
  • n == grid[i].length
  • 1 <= m, n <= 100
  • 1 <= grid[i][j] <= 105

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/get-biggest-three-rhombus-sums-in-a-grid 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。




const getBiggestThree = grid => {
  let m = grid.length, n = grid[0].length, queue = [0, 0, 0]
  const addNum = num => {
    for (let i = 0; i < 3; i++) {
      if (num === queue[i]) return
      if (num < queue[i]) continue
      queue.splice(i, 0, num)
      return queue.pop()
    }
  }
  const computeNum = (i, j, k = 0) => {
    while (++k) {
      let arr = [[-1, 1], [1, 1], [1, -1], [-1, -1]], map = [], sum = 0
      for (let x = 0; x < 4; x++) {
        for (let y = 0; y < k; y++) map.push(x)
      }
      for (let key of map) {
        const [a, b] = arr[key]
        i += a
        j += b
        if (!grid[i] || grid[i][j] === void 0) return
        sum += grid[i][j]
      }
      addNum(sum)
    }
  }
  for (let i = 0; i < m; i++) {
    for (let j = 0; j < n; j++) {
      addNum(grid[i][j])
      computeNum(i, j)
    }
  }
  return queue.filter(n => n)
}

buuing avatar Jun 02 '21 15:06 buuing