-
Notifications
You must be signed in to change notification settings - Fork 0
/
slice.go
125 lines (112 loc) · 2.24 KB
/
slice.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package util
import (
"errors"
"fmt"
"math/rand"
"strconv"
"strings"
"time"
)
// SliceShuffle shuffle a slice
func SliceShuffle(s []any) {
rand.NewSource(time.Now().UnixNano())
rand.Shuffle(len(s), func(i, j int) {
s[i], s[j] = s[j], s[i]
})
}
// SliceReverse slice 反转
func SliceReverse(s []any) {
for i := len(s)/2 - 1; i >= 0; i-- {
opp := len(s) - 1 - i
s[i], s[opp] = s[opp], s[i]
}
}
// InSliceInt 是否在 slice
func InSliceInt(s int, ss []int) bool {
for _, v := range ss {
if v == s {
return true
}
}
return false
}
// InSliceStr 是否在 slice
func InSliceStr(s string, ss []string) bool {
for _, v := range ss {
if v == s {
return true
}
}
return false
}
// SliceDeleteElem 索引删除 slice 元素
func SliceDeleteElem(i int, s []any) ([]any, error) {
if len(s) == 0 {
return nil, errors.New("slice is empty")
}
if i < 0 || i >= len(s) {
return nil, fmt.Errorf("index %d out of range for slice of length %d", i, len(s))
}
s[i] = s[len(s)-1]
s[len(s)-1] = nil
s = s[:len(s)-1]
return s, nil
}
// SliceIntJoin 整型数组拼接成字符串
func SliceIntJoin(s []int, sep string) string {
if len(s) <= 1 {
return ""
}
var str strings.Builder
str.WriteString(strconv.Itoa(s[0]))
for i := 1; i < len(s); i++ {
str.WriteString(sep)
str.WriteString(strconv.Itoa(s[i]))
}
return str.String()
}
// SliceToInt 字符串数组转换成整型数组
func SliceToInt(ss []string) (ii []int) {
for _, s := range ss {
i, _ := strconv.Atoi(s)
ii = append(ii, i)
}
return ii
}
// SliceBigFilter 过滤切片元素 适合大切片
func SliceBigFilter(a []int, f func(v int) bool) []int {
for i := 0; i < len(a); i++ {
if !f(a[i]) {
a = append(a[:i], a[i+1:]...)
i--
}
}
return a
}
// SliceSmallFilter 过滤切片元素 适合小切片
func SliceSmallFilter(a []int, f func(v int) bool) []int {
ret := make([]int, 0, len(a))
for _, val := range a {
if f(val) {
ret = append(ret, val)
}
}
return ret
}
// SliceIntDeduplication 去除重复的元素
func SliceIntDeduplication(a []int) []int {
if a == nil {
return nil
}
m := make(map[int]struct{}, len(a))
j := 0
for _, v := range a {
if _, ok := m[v]; ok {
continue
}
m[v] = struct{}{}
a[j] = v
j++
}
return a[:j]
}