566 finish

Author: aQua

Algorithms/0566.reshape-the-matrix/README.md

@@ -38,10 +38,7 @@ Note:
 1. The height and width of the given matrix is in range [1, 100].
 1. The given r and c are all positive.
 
-
 ## 解题思路
+新旧矩阵元素的位置，存在对应关系。依次填入即可。
 
-
-## 总结
-
-
+## 总结

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

@@ -1,21 +1,19 @@
 package Problem0566
 
 func matrixReshape(nums [][]int, r int, c int) [][]int {
-	length := len(nums) * len(nums[0])
-	if length != r*c {
+	if len(nums) == 0 || len(nums[0]) == 0 || len(nums)*len(nums[0]) != r*c || len(nums) == r && len(nums[0]) == c {
 		return nums
 	}
 
-	// 先降维成[]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]
+	count, col := 0, len(nums[0])
+	for i := range res {
+		res[i] = make([]int, c)
+
+		for j := range res[i] {
+			res[i][j] = nums[count/col][count%col]
+			count++
+		}
 	}
 
 	return res