566 accepted

Author: aQua

Algorithms/0566.reshape-the-matrix/reshape-the-matrix.go

@@ -1,6 +1,22 @@
 package Problem0566
 
 func matrixReshape(nums [][]int, r int, c int) [][]int {
+	length := len(nums) * len(nums[0])
+	if length != r*c {
+		return nums
+	}
 
-	return nil
+	// 先降维成[]int
+	temp := make([]int, 0, r*c)
+	for i := range nums {
+		temp = append(temp, nums[i]...)
+	}
+
+	// 再升维
+	res := make([][]int, r)
+	for i := 0; i < r; i++ {
+		res[i] = temp[i*c : (i+1)*c]
+	}
+
+	return res
 }

Algorithms/0566.reshape-the-matrix/reshape-the-matrix_test.go

@@ -30,6 +30,17 @@ func Test_Problem0566(t *testing.T) {
 
 	qs := []question{
 
+		question{
+			para{[][]int{
+				[]int{1, 2},
+				[]int{3, 4},
+			}, 2, 4},
+			ans{[][]int{
+				[]int{1, 2},
+				[]int{3, 4},
+			}},
+		},
+
 		question{
 			para{[][]int{
 				[]int{1, 2},
@@ -40,6 +51,18 @@ func Test_Problem0566(t *testing.T) {
 			}},
 		},
 
+		question{
+			para{[][]int{
+				[]int{1, 2},
+				[]int{3, 4},
+			}, 4, 1},
+			ans{[][]int{
+				[]int{1},
+				[]int{2},
+				[]int{3},
+				[]int{4},
+			}},
+		},
 		// 如需多个测试，可以复制上方元素。
 	}