按照矩阵相乘的法则,先判断是否符合相乘的条件
func matrixReshape(nums [][]int, r int, c int) [][]int {
ret := make([][]int, r)
for k := range ret {
ret[k] = make([]int, c)
}
if len(nums) == 0 || r*c != len(nums)*len(nums[0]) {
return nums
}
rows, cols := 0, 0
for i := 0; i < len(nums); i++ {
for j := 0; j < len(nums[0]); j++ {
ret[rows][cols] = nums[i][j]
cols++
if cols == c {
rows++
cols = 0
}
}
}
return ret
}
更多内容请移步我的repo:https://github.com/anakin/golang-leetcode