Go语言中高效移除切片多项元素的策略与实践

type Record struct {
    id   int
    name string
}

// deleteRecords 原地移除切片中的指定记录，并保持剩余元素的相对顺序。
// 适用于待移除ID列表较小（例如40个以内）的场景。
func deleteRecords(data []*Record, ids []int) []*Record {
    w := 0 // 写入指针，指向下一个要写入的位置

loop: // 标签，用于跳出内部循环后直接进入外部循环的下一次迭代
    for _, x := range data {
        // 检查当前元素x的ID是否在待删除列表中
        for _, id := range ids {
            if id == x.id {
                continue loop // 如果匹配，跳过当前元素，继续外层循环的下一次迭代
            }
        }
        // 如果当前元素x的ID不在待删除列表中，则保留它
        data[w] = x
        w++
    }
    // 返回截取后的切片，其长度为w
    return data[:w]
}

// reorder 原地移除切片中的指定记录，不保证剩余元素的相对顺序。
// 在对顺序无要求时，此方法通常比保持顺序的方法更快。
func reorder(data []*Record, ids []int) []*Record {
    n := len(data) // 当前有效元素的数量
    i := 0         // 读取指针

loop:
    for i < n {
        r := data[i]
        // 检查当前元素r的ID是否在待删除列表中
        for _, id := range ids {
            if id == r.id {
                // 如果匹配，将当前元素与切片末尾元素交换，然后缩短切片长度
                data[i] = data[n-1]
                n--             // 有效元素数量减少
                continue loop   // 继续外层循环的下一次迭代，重新检查当前i位置的新元素
            }
        }
        i++ // 如果当前元素不需要删除，则移动到下一个元素
    }
    // 返回截取后的切片，其长度为n
    return data[0:n]
}

// deletePreserve 将符合条件的记录复制到一个新切片中，保持原切片不变。
// 适用于需要保留原始数据或构建全新结果集的场景。
func deletePreserve(data []*Record, ids []int) []*Record {
    wdata := make([]*Record, len(data)) // 创建一个新切片，初始容量与原切片相同
    w := 0 // 写入指针

loop:
    for _, x := range data {
        // 检查当前元素x的ID是否在待删除列表中
        for _, id := range ids {
            if id == x.id {
                continue loop // 如果匹配，跳过当前元素，继续外层循环的下一次迭代
            }
        }
        // 如果当前元素x的ID不在待删除列表中，则复制到新切片中
        wdata[w] = x
        w++
    }
    // 返回截取后的新切片
    return wdata[0:w]
}

// deleteRecordsOptimized 优化后的原地移除方法，使用哈希表加速ID查找。
// 适用于待移除ID列表较大的场景。
func deleteRecordsOptimized(data []*Record, ids []int) []*Record {
    // 构建一个哈希表，用于快速查找待删除的ID
    idMap := make(map[int]struct{}, len(ids))
    for _, id := range ids {
        idMap[id] = struct{}{}
    }

    w := 0 // 写入指针
    for _, x := range data {
        // 使用哈希表进行查找，时间复杂度接近O(1)
        if _, found := idMap[x.id]; !found {
            data[w] = x
            w++
        }
    }
    return data[:w]
}