diff --git a/src/main/java/MainExercises.java b/src/main/java/MainExercises.java index 2ede0f1..79f6b46 100644 --- a/src/main/java/MainExercises.java +++ b/src/main/java/MainExercises.java @@ -78,6 +78,8 @@ public class MainExercises int rows = matrix.length; int cols = matrix[0].length; + if(rows == 1) return matrix[0]; + int[] result = new int[rows * cols]; int top = 0, bottom = rows - 1, left = 0, right = cols - 1;