如何在Java中实现多维数组的自定义迭代器？

import java.util.Iterator;

public class MultiDimensionalArray<T> implements Iterable<T> {
    private T[][] array;

    public MultiDimensionalArray(T[][] array) {
        this.array = array;
    }

    public Iterator<T> iterator() {
        return new CustomIterator<>(array);
    }

    private class CustomIterator<T> implements Iterator<T> {
        private T[][] array;
        private int row;
        private int col;

        public CustomIterator(T[][] array) {
            this.array = array;
            row = 0;
            col = 0;
        }

        public boolean hasNext() {
            return row < array.length && col < array[row].length;
        }

        public T next() {
            T element = array[row][col++];
            if (col >= array[row].length) {
                row++;
                col = 0;
            }
            return element;
        }

        public void remove() {
            // 可选择性地实现
        }
    }
}

public class Main {
    public static void main(String[] args) {
        Integer[][] array = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
        MultiDimensionalArray<Integer> multiArray = new MultiDimensionalArray<>(array);

        Iterator<Integer> iterator = multiArray.iterator();
        while (iterator.hasNext()) {
            System.out.println(iterator.next());
        }
    }
}