-
Notifications
You must be signed in to change notification settings - Fork 3
/
remove.go
127 lines (106 loc) · 2.75 KB
/
remove.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
126
127
package sliceKit
// RemoveByIndex 根据下标删除指定元素.
/*
PS:
(0) 也可以用于移除"第一个"或"最后一个"个元素;
(1) 为什么要有两个返回值且第一个是slice?因为 append() 并不会修改传参slice,返回的是一个新的slice.
(2) 调用此方法后,必须第一时间将 第一个返回值 赋值给 传进来的那个slice变量!!!
@param s (1) 可以为nil
(2) 不会修改传参s
@param index 下标(索引),取值范围: [0, length)
e.g.
s := []int{0, 1, 2, 3}
s1, item, ok := sliceKit.RemoveByIndex(s, 2)
fmt.Println(s) // [0 1 2 3]
fmt.Println(s1) // [0 1 3]
fmt.Println(item) // 2
fmt.Println(ok) // true
*/
func RemoveByIndex[T any](s []T, index int) (s1 []T, item T, ok bool) {
if len(s) == 0 {
s1 = s
return
}
item = s[index]
// !!!: 下面一行代码执行后,会修改外部的slice
//s1 = append(s[:index], s[index+1:]...)
s1 = append(s1, s[:index]...)
s1 = append(s1, s[index+1:]...)
ok = true
return
}
// RemoveFirst 移除第一个元素(如果有的话)
/*
@param s 可以为nil
e.g.
[int](nil) => [] 0 false
([]int{0}) => [] 0 true
([]int{0, 1}) => [1] 0 true
*/
func RemoveFirst[T any](s []T) (s1 []T, item T, ok bool) {
if len(s) == 0 {
s1 = s
return
}
// 此时第一个返回值 必定不为nil && len >= 0
item = s[0]
s1 = s[1:]
ok = true
return
}
// RemoveLast 移除最后一个元素(如果有的话)
/*
@param s 可以为nil
e.g.
[int](nil) => [] 0 false
([]int{0}) => [] 0 true
([]int{0, 1}) => [0] 1 true
*/
func RemoveLast[T any](s []T) (s1 []T, item T, ok bool) {
if len(s) == 0 {
s1 = s
return
}
// 此时第一个返回值 必定不为nil && len >= 0
index := len(s) - 1
item = s[index]
s1 = s[0:index]
ok = true
return
}
// Remove 移除元素.
/*
PS:
(1) 切片实例s中,存在多个item的话,仅会移除第一个.
@param s (1)可以为nil (2)不会修改传参s
e.g. 反例
texts := []string{"0", "1", "2"}
fmt.Println(texts) // [0 1 2]
texts1, _ := sliceKit.Remove(texts, "1")
fmt.Println(texts) // [0 2 2]
fmt.Println(texts1) // [0 2]
*/
func Remove[T comparable](s []T, item T) ([]T, bool) {
index := IndexOf(s, item)
if index == -1 {
// 如果元素不存在于切片中,返回原始切片
return s, false
}
s, _, ok := RemoveByIndex(s, index)
return s, ok
}
// RemoveBy 移除不满足条件的元素(返回的是一个新的slice实例).
/*
@param s 可以为nil
@param predicate (1)不能为nil (2)返回值为true: 移除当前元素
@return (1)非nil (2)len>=0
*/
func RemoveBy[T comparable](s []T, predicate func(element T) bool) []T {
result := make([]T, 0, len(s))
for _, element := range s {
if !predicate(element) {
result = append(result, element)
}
}
return result
}