refactor: change return from String to List

Author: alxkm

src/main/java/com/thealgorithms/others/FloydTriangle.java

@@ -1,5 +1,8 @@
 package com.thealgorithms.others;
 
+import java.util.ArrayList;
+import java.util.List;
+
 final class FloydTriangle {
     private FloydTriangle() {
     }
@@ -8,21 +11,20 @@ private FloydTriangle() {
      * Generates a Floyd Triangle with the specified number of rows.
      *
      * @param rows The number of rows in the triangle.
-     * @return A string representing the Floyd Triangle.
+     * @return A List representing the Floyd Triangle.
      */
-    public static String generateFloydTriangle(int rows) {
-        StringBuilder triangle = new StringBuilder();
+    public static List<List<Integer>> generateFloydTriangle(int rows) {
+        List<List<Integer>> triangle = new ArrayList<>();
         int number = 1;
 
         for (int i = 0; i < rows; i++) {
+            List<Integer> row = new ArrayList<>();
             for (int j = 0; j <= i; j++) {
-                triangle.append(number++).append(" ");
-            }
-            if (i < rows - 1) {
-                triangle.append("\n");
+                row.add(number++);
             }
+            triangle.add(row);
         }
 
-        return triangle.toString();
+        return triangle;
     }
 }

src/test/java/com/thealgorithms/others/FloydTriangleTest.java

@@ -1,36 +1,42 @@
 package com.thealgorithms.others;
 
+import org.junit.jupiter.api.Test;
+
 import static org.junit.jupiter.api.Assertions.assertEquals;
 
-import org.junit.jupiter.api.Test;
+import java.util.Arrays;
+import java.util.List;
 
 public class FloydTriangleTest {
 
     @Test
     public void testGenerateFloydTriangleWithValidInput() {
-        String expectedOutput = "1 \n2 3 \n4 5 6 ";
+        List<List<Integer>> expectedOutput = Arrays.asList(Arrays.asList(1), Arrays.asList(2, 3), Arrays.asList(4, 5, 6));
         assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(3));
     }
 
     @Test
     public void testGenerateFloydTriangleWithOneRow() {
-        String expectedOutput = "1 ";
+        List<List<Integer>> expectedOutput = Arrays.asList(Arrays.asList(1));
         assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(1));
     }
 
     @Test
     public void testGenerateFloydTriangleWithZeroRows() {
-        assertEquals("", FloydTriangle.generateFloydTriangle(0));
+        List<List<Integer>> expectedOutput = Arrays.asList();
+        assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(0));
     }
 
     @Test
     public void testGenerateFloydTriangleWithNegativeRows() {
-        assertEquals("", FloydTriangle.generateFloydTriangle(-3));
+        List<List<Integer>> expectedOutput = Arrays.asList();
+        assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(-3));
     }
 
     @Test
     public void testGenerateFloydTriangleWithMultipleRows() {
-        String expectedOutput = "1 \n2 3 \n4 5 6 \n7 8 9 10 \n11 12 13 14 15 ";
+        List<List<Integer>> expectedOutput = Arrays.asList(Arrays.asList(1), Arrays.asList(2, 3), Arrays.asList(4, 5, 6), Arrays.asList(7, 8, 9, 10), Arrays.asList(11, 12, 13, 14, 15));
         assertEquals(expectedOutput, FloydTriangle.generateFloydTriangle(5));
     }
 }
+