需求:给定一个数组和一个正整数,要求把数组分割成多个正整数大小的数组,如果不够分,则最后一个数组分到剩余的所有元素。
示例1:
数组:[1, 2, 3, 4, 5, 6, 7, 8, 9, 10],正整数:2
期望结果: [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
示例2:
数组:[1, 2, 3, 4, 5, 6, 7, 8, 9],正整数:2
期望结果: [[1, 2], [3, 4], [5, 6], [7, 8], [9]]
下面是我的实现代码:
//分割数组,根据传入的数组和分割大小,将数组分割为大小等于指定大小的多个数组,如果不够分,则最后一个数组元素小于其他数组 func splitArray(arr []int, num int64) [][]int { max := int64(len(arr)) //判断数组大小是否小于等于指定分割大小的值,是则把原数组放入二维数组返回 if max <= num { return [][]int{arr} } //获取应该数组分割为多少份 var quantity int64 if max%num == 0 { quantity = max / num } else { quantity = (max / num) + 1 } //声明分割好的二维数组 var segments = make([][]int, 0) //声明分割数组的截止下标 var start, end, i int64 for i = 1; i <= quantity; i++ { end = i*num if i != quantity { segments = append(segments, arr[start:end]) } else { segments = append(segments, arr[start:]) } start = i*num } return segments }
附:这个是我在实际业务需求中遇到的问题,不知道大家有没有更好写法
您可能感兴趣的文章: