1030. 距离顺序排列矩阵单元格

2021/12/19 6:21:54

本文主要是介绍1030. 距离顺序排列矩阵单元格,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

给出 R 行 C 列的矩阵,其中的单元格的整数坐标为 (r, c),满足 0 <= r < R 且 0 <= c < C。

另外,我们在该矩阵中给出了一个坐标为 (r0, c0) 的单元格。

返回矩阵中的所有单元格的坐标,并按到 (r0, c0) 的距离从最小到最大的顺序排,其中,两单元格(r1, c1) 和 (r2, c2) 之间的距离是曼哈顿距离,|r1 - r2| + |c1 - c2|。(你可以按任何满足此条件的顺序返回答案。)

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

import java.util.HashSet;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Set;

class Solution {

    private int[] dx = {0, 1, 0, -1};

    private int[] dy = {1, 0, -1, 0};

    private int hash(int x, int y) {
        return x * 1000 + y;
    }

    public int[][] allCellsDistOrder(int rows, int cols, int rCenter, int cCenter) {
        if (rows <= 0 || cols <= 0) {
            return new int[0][0];
        }

        Set<Integer> visited = new HashSet<>();
        Queue<int[]> queue = new LinkedList<>();
        int[][] ret = new int[rows * cols][2];
        int idx = 0;
        queue.offer(new int[]{rCenter, cCenter});
        visited.add(hash(rCenter, cCenter));
        while (!queue.isEmpty()) {
            int[] node = queue.poll();
            ret[idx++] = node;

            for (int i = 0; i < 4; ++i) {
                int x = dx[i] + node[0];
                int y = dy[i] + node[1];
                int hash = hash(x, y);
                if (x >= 0 && x < rows && y >= 0 && y < cols && !visited.contains(hash)) {
                    queue.offer(new int[]{x, y});
                    visited.add(hash);
                }
            }
        }

        return ret;
    }
}


这篇关于1030. 距离顺序排列矩阵单元格的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程