golang多个切片合并后按照指定长度进行切割分组

package main

import (
	"errors"
	"fmt"
)

func main() {
	fmt.Println(ChunkSlicesGroup(7, false, []int{1, 2, 3, 4}, []int{1, 22, 31, 42}, []int{55, 66})) //[[1 2 3 4 1 22 31] [42 55 66]] <nil>

    fmt.Println(ChunkSlicesGroup(7, true, []int{1, 2, 3, 4}, []int{1, 22, 31, 42}, []int{55, 66})) //[[1 2 3 4 22 31 42] [55 66]] <nil>
}

// ChunkSlicesGroup 多个切片合并后按照指定长度进行切割分组
// Size 每个被分割的切片组长度
// IsDeduplication 是否需要去重切片元素
// Slices 需要处理的切片
func ChunkSlicesGroup[T comparable](Size int, IsDeduplication bool, Slices ...[]T) ([][]T, error) {
	if Size <= 0 {
		return nil, errors.New("size is Zero")
	}

	uniqueElements := make([]T, 0)
	if IsDeduplication {
		deduplicationMap := make(map[T]struct{})
		for _, slices := range Slices {
			for _, element := range slices {
				if _, exists := deduplicationMap[element]; !exists {
					deduplicationMap[element] = struct{}{}
					uniqueElements = append(uniqueElements, element)
				}
			}
		}
	} else {
		for _, slices := range Slices {
			for _, element := range slices {
				uniqueElements = append(uniqueElements, element)
			}
		}
	}

	uniqueElementsLen := len(uniqueElements)
	slicesGroup := make([][]T, 0, (uniqueElementsLen+Size-1)/Size)

	for i := 0; i < uniqueElementsLen; i += Size {
		end := i + Size
		if end > uniqueElementsLen {
			end = uniqueElementsLen
		}
		slicesGroup = append(slicesGroup, uniqueElements[i:end])
	}
	return slicesGroup, nil
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值