我想在指定值之前剪切一個整數數組，并返回一個包含這些值和數組中剩余值的數組。我們可以假設數組已排序。這是我到目前為止所擁有的：func cutIntArrBefore(arr1 []int, n int) ([]int, []int) { arr2 := make([]int, 0, len(arr1)) sliceIndex := 0 for i, num := range arr1 { if num < n { arr2 = append(arr2, num) sliceIndex = i } } sliceIndex = sliceIndex + 1 if sliceIndex >= len(arr1) { return arr2, nil } else { arr1 := arr1[sliceIndex:] return arr2, arr1 }}測試代碼：func main() { var arr1, arr2, arr3 []int arr1 = []int{1,2,3,4,5,6,7,8} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{1,5} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{1} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{5} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{5,6} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{5,5} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3) arr1 = []int{7,7,7} arr2, arr3 = cutIntArrBefore(arr1, 5) fmt.Printf("(%+v) = %+v, %+v\n", arr1, arr2, arr3)}輸出：([1 2 3 4 5 6 7 8]) = [1 2 3 4], [5 6 7 8]([1 5]) = [1], [5]([1]) = [1], []([5]) = [], []([5 6]) = [], [6]([5 5]) = [], [5]([7 7 7]) = [], [7 7]不幸的是，如您所見，如果第一個元素在指定值之后，它將被跳過。我想盡可能優雅地做到這一點。我希望有另一種方法，而不必創建兩個數組或添加另一個 if 語句。
1 回答
拉風的咖菲貓
TA貢獻1995條經驗 獲得超2個贊
arr[:i]您的實現包含一個差一錯誤，可以通過使用切片表達式和將最左邊的目標值的索引作為軸心點輕松解決arr[i:]。
另外，考慮使用sort.SearchInts(...)to find the target index in?O(lg(n))time 而不是O(n).?使用內置函數也可能會提高代碼的易讀性和可維護性。
例如（去游樂場）：
func cutBefore(xs []int, x int) ([]int, []int) {
? i := sort.SearchInts(xs, x)
? return xs[:i], xs[i:]
}
func main() {
? xss := [][]int{
? ? {1, 2, 3, 4, 5, 6, 7, 8},
? ? {1, 5},
? ? {1},
? ? {5},
? ? {5, 6},
? ? {5, 5},
? ? {7, 7, 7},
? }
? for _, xs := range xss {
? ? fmt.Println(cutBefore(xs, 5))
? }
? // [1 2 3 4] [5 6 7 8]
? // [1] [5]
? // [1] []
? // [] [5]
? // [] [5 6]
? // [] [5 5]
? // [] [7 7 7]
}
- 1 回答
- 0 關注
- 125 瀏覽
添加回答
舉報
0/150
提交
取消