Java通过2D数组迭代

时间:2013-03-24 21:51:52

标签: java arrays multidimensional-array iteration

我有一个像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

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

我的理论思路是按顺序迭代每个方向。例如,从上升开始,所以我会检查2

以上的所有值
1
1
2

由于我没有找到任何零,请检查右上角

   1
  1
 2

仍然没有0,所以走到右边。一找到我的第一个0,就打破。

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

但我是java的新手,所以我不知道最好的方法。有什么想法吗?

1 个答案:

答案 0 :(得分:1)

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

在您的场景中找到第一个零将涉及遍历每个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();
    }
}