写点什么

[Day29]-[数组] 将一维数组转变成二维数组

作者:方勇(gopher)
  • 2022 年 4 月 28 日
  • 本文字数:1100 字

    阅读完需:约 4 分钟

2022. 将一维数组转变成二维数组

给你一个下标从 0 开始的一维整数数组 original 和两个整数 m 和  n 。你需要使用 original 中 所有 元素创建一个 m 行 n 列的二维数组。

original 中下标从 0 到 n - 1 (都 包含 )的元素构成二维数组的第一行,下标从 n 到 2 * n - 1 (都 包含 )的元素构成二维数组的第二行,依此类推。

请你根据上述过程返回一个 m x n 的二维数组。如果无法构成这样的二维数组,请你返回一个空的二维数组。

 

示例 1:


输入:original = [1,2,3,4], m = 2, n = 2输出:[[1,2],[3,4]]解释:构造出的二维数组应该包含 2 行 2 列。original 中第一个 n=2 的部分为 [1,2] ,构成二维数组的第一行。original 中第二个 n=2 的部分为 [3,4] ,构成二维数组的第二行。
复制代码

示例 2:

输入:original = [1,2,3], m = 1, n = 3输出:[[1,2,3]]解释:构造出的二维数组应该包含 1 行 3 列。将 original 中所有三个元素放入第一行中,构成要求的二维数组。
复制代码

示例 3:

输入:original = [1,2], m = 1, n = 1输出:[]解释:original 中有 2 个元素。无法将 2 个元素放入到一个 1x1 的二维数组中,所以返回一个空的二维数组。
复制代码

示例 4:

输入:original = [3], m = 1, n = 2输出:[]解释:original 中只有 1 个元素。无法将 1 个元素放满一个 1x2 的二维数组,所以返回一个空的二维数组。
复制代码

题解:

func TestConstruct2DArray(t *testing.T) {	var construct2DArray func(original []int, m int, n int) [][]int	construct2DArray = func(original []int, m, n int) [][]int {		length := len(original)		if length != m*n {			return nil		}		var res = make([][]int, m)		for i := 0; i < m; i++ {			res[i] = make([]int, n)		}		for i := 1; i <= m; i++ {			index := 0			for j := (i - 1) * n; j <= i*n-1; j++ {				res[i-1][index] = original[j]				index++			}		}		return res	}
t.Log(construct2DArray([]int{1, 2, 3, 4}, 2, 2))}
func TestConstruct2DArray2(t *testing.T) { var construct2DArray func(original []int, m int, n int) [][]int construct2DArray = func(original []int, m, n int) [][]int { length := len(original) if length != m*n { return nil } // var res = make([][]int, 0, m) // for i := 0; i < length; i += n { // res = append(res, original[i:i+n]) // } var res = make([][]int, m) for i := 0; i < m; i++ { res[i] = original[i*n : (i+1)*n] } return res }
t.Log(construct2DArray([]int{1, 2, 3, 4}, 2, 2))}
复制代码


用户头像

Dead or Alive. 生存战斗是知识的源泉! 2018.11.08 加入

我是一名SRE哨兵,目前是好大夫基础架构部高级工程师。专注于 SRE,微服务、中间件的稳定性和可用性建设,整体负责好大夫服务治理云平台的设计和搭建!

评论

发布
暂无评论
[Day29]-[数组]将一维数组转变成二维数组_LeetCode_方勇(gopher)_InfoQ写作社区