Java 2d数组迭代

Question

我在Java中有一个2D数组，像这样:

int 2d_arr[5][5];

举个例子，黑板是这样的:

1 1 1 1 1
0 1 1 1 1
1 1 2 1 0
0 1 1 1 1
0 0 1 1 1

从第三行中的2开始，我希望能够在每个方向上移动(上、下、左、右和对角线)。在代码中，我如何遍历数组的每个方向，直到我找到一个0?

我的理论想法是按顺序在每个方向上迭代。例如，从向上开始，所以我会检查2

上面的所有值

1
1
2

由于我没有找到任何零，检查右上方对角线

   1
  1
 2

仍然没有0，所以往右转。只要找到第一个0，就中断。

尝试:我实际上知道如何做到这一点与一堆if和for循环，但我正在寻找一种方法来编辑该代码到一个更简单，更容易阅读的版本

但是我是java新手，所以我不知道最好的方法。什么好主意吗?

Answer 1

一个TwoD迭代器显然是一个很好的开始。我希望你能毫不费力地自己完成剩下的工作。

在您的场景中找到第一个零将涉及遍历每个Direction并沿着该方向遍历整个板，直到迭代结束或您找到您的零。

螺旋搜索包括在每个方向上启动一个迭代器，依次执行一步并检查每个迭代器，直到其中一个返回一个可以找到零的点。

public class TwoDIteratorTest {
    // An ubiquitous Point class
    static class Point {
        final int x;
        final int y;
        public Point(int x, int y) {
            this.x = x;
            this.y = y;
        }
        @Override
        public String toString() {
            return "{" + x + "," + y + "}";
        }
    }
    // All possible directions.
    enum Direction {
        North(0, 1),
        NorthEast(1, 1),
        East(1, 0),
        SouthEast(1, -1),
        South(0, -1),
        SouthWest(-1, -1),
        West(-1, 0),
        NorthWest(-1, 1);
        private final int dx;
        private final int dy;
        Direction(int dx, int dy) {
            this.dx = dx;
            this.dy = dy;
        }
        // Step that way
        public Point step(Point p) {
            return new Point(p.x + dx, p.y + dy);
        }
    }
    static class TwoDIterator implements Iterable<Point> {
        // Where we are now
        Point i;
        // Direction to move in.
        private final Direction step;
        // Limits.
        private final Point min;
        private final Point max;
        // Next position to go to.
        Point next = null;
        // Normal constructor.
        public TwoDIterator(Point start, Direction step, Point min, Point max) {
            i = next = start;
            this.step = step;
            this.min = min;
            this.max = max;
        }
        // Some simplified constructors
        public TwoDIterator(int x, int y, Direction step, Point min, Point max) {
            this(new Point(x, y), step, min, max);
        }
        public TwoDIterator(int x, int y, Direction step, int minx, int miny, int maxx, int maxy) {
            this(new Point(x, y), step, new Point(minx, miny), new Point(maxx, maxy));
        }
        // The iterator.
        @Override
        public Iterator<Point> iterator() {
            return new Iterator<Point>() {
                // hasNext calculates next if necessary and checks it against the stabliched limits.
                @Override
                public boolean hasNext() {
                    if (next == null) {
                        // Step one.
                        next = step.step(i);
                        // Stop at limits.
                        if (next.x < min.x
                                || next.x > max.x
                                || next.y < min.y || next.y > max.y) {
                            next = null;
                        }
                    }
                    return next != null;
                }
                @Override
                public Point next() {
                    if (hasNext()) {
                        // Make our move.
                        i = next;
                        next = null;
                        return i;
                    }
                    return null;
                }
                @Override
                public void remove() {
                    throw new UnsupportedOperationException("Not supported.");
                }
            };
        }
    }
    public void test() {
        // Test all directions.
        for (Direction d : Direction.values()) {
            System.out.print(d + " - ");
            for (Point p : new TwoDIterator(0, 0, d, -5, -5, 5, 5)) {
                System.out.print(p + ",");
            }
            System.out.println();
        }
    }
    public static void main(String[] args) throws InterruptedException {
        new TwoDIteratorTest().test();
    }
}