Создать метод получения массива

class Flat {
    private int square = 50; 
    private int room = 2;
 
    public Flat(int square, int room) {
        this.square = square;
        this.room = room;
    }
 
    public Flat(int square) { 
        this.square = square;
    }
 
    public Flat() {
    }
 
    public int getSquare() {
        return square;
    }
 
    public int getRoom() {
        return room;
    }
 
    public void setSquare(int square) {
        this.square = square;
    }
 
    public void setRoom(int room) {
        this.room = room;
    }
}
class DwellingFloor {
    private int cnt=7;
    private Flat flats[] = new Flat[cnt];                               //класс основанный на массиве flats 
 
    public DwellingFloor(int cnt) {
        this.flats = new Flat[cnt];
    } //конструктор принимает количество квартир
 
    public int getFlatsCount() {
        return flats.length;
    }                 //метод получения количества квартир на этаже
 
    public int getAllSquare() {
        for (int i = 0; i < flats.length; i++) {
            flats[i] = new Flat();
        }
        int sum = 0;
        for (int k = 0; k < flats.length; k++)
            sum += flats[k].getSquare();
        return sum;
    }                            //метод получения общей площади квартир
 
    public int getAllRoom() {
        for (int i = 0; i < flats.length; i++) {
            flats[i] = new Flat();
        }
        int sum = 0;
        for (int k = 0; k < flats.length; k++)
            sum += flats[k].getRoom();
        return sum;
    }                              //метод получения общего количества комнат
 
    public Flat getArrayFlats() {
        for (Flat val : flats) {
            System.out.println(+val.getRoom() + " " + val.getSquare());
    }
        return null;
    }                          //метод получения массива квартир
 
}

class DwellingFloor {
    private int cnt=7;
    private Flat flats[];
    
    public DwellingFloor(Flat someFlats[]) {
        flats = Arrays.copyOf(someFlats, someFlats.length);
    }
 
    Flat[] getFlats() {
        return flats;
    }
    
    Flat getFlat(int flatNumber) {
        return flats[flatNumber];
    }
. . .

        DwellingFloor d = new DwellingFloor(new Flat[]{new Flat(90, 3), new Flat(36, 1), new Flat(60, 2)});
        System.out.println(d.getFlat(1).getRoom());
        System.out.println(d.getFlats()[2].getRoom());