C#|.net core 基础 - 扩展数组添加删除性能最好的方法

今天在编码的时候遇到了一个问题，需要对数组变量添加新元素和删除元素，因为数组是固定大小的，因此对新增和删除并不友好，但有时候又会用到，因此想针对数组封装两个扩展方法：新增元素与删除元素，并能到达以下三个目标：

1、性能优异；

2、兼容性好；

3、方便使用；

这三个目标最麻烦的应该就是性能优异了，比较后面两个可以通过泛型方法，扩展方法，按引用传递等语法实现，性能优异却要在十来种实现方法中选出两个最优的实现。那关于数组新增和删除元素你能想到多少种实现呢？下面我们来一起看看那个性能最好。

01、新增元素实现方法对比

1、通过List方法实现

通过转为List，再用AddRange方法添加元素，最后再转为数组返回。代码实现如下：

public static int[] AddByList(int[] source, int[] added)
{
    var list = source.ToList();
    list.AddRange(added);
    return list.ToArray();
}

2、通过IEnumerable方法实现

因为数组实现了IEnumerable接口，所以可以直接调用Concat方法实现两个数组拼接。代码实现如下：

public static int[] AddByConcat(int[] source, int[] added)
{
    return source.Concat(added).ToArray();
}

3、通过Array方法实现

Array有个Copy静态方法可以实现把数组复制到目标数组中，因此我们可以先构建一个大数组，然后用Copy方法把两个数组都复制到大数组中。代码实现如下：

public static int[] AddByCopy(int[] source, int[] added)
 {
     var size = source.Length + added.Length;
     var array = new int[size];
     // 复制原数组
     Array.Copy(source, array, source.Length);
     // 添加新元素
     Array.Copy(added, 0, array, source.Length, added.Length);
     return array;
 }

4、通过Span方法实现

Span也有一个类似Array的Copy方法，功能也类似，就是CopyTo方法。代码实现如下：

public static int[] AddBySpan(int[] source, int[] added)
{
    Span<int> sourceSpan = source;
    Span<int> addedSpan = added;
    Span<int> span = new int[source.Length + added.Length];
    // 复制原数组
    sourceSpan.CopyTo(span);
    // 添加新元素
    addedSpan.CopyTo(span.Slice(sourceSpan.Length));
    return span.ToArray();
}

我想到了4种方法来实现，如果你有不同的方法希望可以给我留言，不吝赐教。那么那种方法效率最高呢？按我理解作为现在.net core性能中的一等公民Span应该性能是最好的。

我们也不瞎猜了，直接来一组基准测试对比。我们对4个方法，分三组测试，每组分别随机生成两个100、1000、10000个元素的数组，然后每组再进行10000次测试。

测试结果如下：

整体排名：AddByCopy > AddByConcat > AddBySpan > AddByList。

可以发现性能最好的竟然是Array的Copy方法，不但速度最优，而且内存使用方面也是最优的。

而我认为性能最好的Span整体表现还不如IEnumerable的Concat方法。

最终Array的Copy方法完胜。

02、删除元素实现方法对比

1、通过List方法实现

还是先把数组转为List，然后再用RemoveAll进行删除，最后把结果转为数组返回。代码实现如下：

public static int[] RemoveByList(int[] source, int[] added)
{
    var list = source.ToList();
    list.RemoveAll(x => added.Contains(x));
    return list.ToArray();
}

2、通过IEnumerable方法实现

因为数组实现了IEnumerable接口，所以可以直接调用Where方法进行过滤。代码实现如下：

public static int[] RemoveByWhere(int[] source, int[] added)
{
     return source.Where(x => !added.Contains(x)).ToArray();
}

3、通过Array方法实现

Array有个FindAll静态方法可以实现根据条件查找数组。代码实现如下：

public static int[] RemoveByArray(int[] source, int[] added)
{
    return Array.FindAll(source, x => !added.Contains(x));
}

4、通过For+List方式实现

直接遍历原数组，把满足条件的元素放入List中，然后转为数组返回。代码实现如下：

public static int[] RemoveByForList(int[] source, int[] added)
{
    var list = new List<int>();
    foreach (int item in source)
    {
        if (!added.Contains(item))
        {
            list.Add(item);
        }
    }
    return list.ToArray();
}

5、通过For+标记+Copy方式实现

还是直接遍历原数组，但是我们不创建新集合，直接把满足的元素放在原数组中，因为从原数组第一个元素迭代，如果元素满足则放入第一个元素其索引自动加1，如果不满足则等下一个满足的元素放入其索引保持不变，以此类推，直至所有元素处理完成，最后再把原数组中满足要求的数组复制到新数据中返回。代码实现如下：

public static int[] RemoveByForMarkCopy(int[] source, int[] added)
{
    var idx = 0;
    foreach (var item in source)
    {
        if (!added.Contains(item))
        {
            // 标记有效元素
            source[idx++] = item;
        }
    }
    // 创建新数组并复制有效元素
    var array = new int[idx];
    Array.Copy(source, array, idx);
    return array;
}

6、通过For+标记+Resize方式实现

这个方法和上一个方法实现基本一致，主要差别在最后一步，这个方法是直接通过Array的Resize静态方法把原数组调整为我们要的并返回。代码实现如下：

public static int[] RemoveByForMarkResize(int[] source, int[] added)
{
    var idx = 0;
    foreach (var item in source)
    {
        if (!added.Contains(item))
        {
            //标记有效元素
            source[idx++] = item;
        }
    }
    //调整数组大小
    Array.Resize(ref source, idx);
    return source;
}

同样的我们再做一组基准测试对比，结果如下：

可以发现最后两个方法随着数组元素增加性能越来越差，而其他四种方法相差不大。既然如此我们就选择Array原生方法FindAll。

03、实现封装方法

新增删除的两个方法已经确定，我们第一个目标就解决了。

既然要封装为公共的方法，那么就必要要有良好的兼容性，我们示例虽然都是用的int类型数组，但是实际使用中不知道会碰到什么类型，因此最好方式是选择泛型方法。这样第二个目标就解决了。

那么第三个目标方便使用要怎么办呢？第一想法既然做成公共方法了，直接做一个帮助类，比如ArrayHelper，然后把两个实现方法直接以静态方法放进去。

但是我更偏向使用扩展方法，原因有二，其一可以利用编辑器直接智能提示出该方法，其二代码更简洁。形如下面两种形式，你更喜欢那种？

//扩展方法
var result = source.Add(added);
//静态帮助类方法
var result = ArrayHelper.Add(source, added);

现在还有一个问题，这个方法是以返回值的方式返回最后的结果呢？还是直接修改原数组呢？两种方式各有优点，返回新数组，则原数组不变便于链式调用也避免一些副作用，直接修改原数组内存效率高。

我们的两个方法是新增元素和删除元素，其语义更贴合对原始数据进行操作其结果也作用在自身。因此我更倾向无返回值的方式。

那现在有个尴尬的问题，不知道你还记得我们上一章节《C#|.net core 基础 - 值传递 vs 引用传递》讲的值传递和引用传递，这里就有个这样的问题，如果我们现在想用扩展方法并且无返回值直接修改原数组，那么需要对扩展方法第一个参数使用ref修饰符，但是扩展方法对此有限制要求【第一个参数必须是struct 或是被约束为结构的泛型类型】，显示泛型数组不满足这个限制。因此无法做到我心目中最理想的封装方式了，下面看看扩展方法和帮助类的代码实现，可以按需使用吧。

public static class ArrayExtensions
{
    public static T[] AddRange<T>(this T[] source, T[] added)
    {
        var size = source.Length + added.Length;
        var array = new T[size];
        Array.Copy(source, array, source.Length);
        Array.Copy(added, 0, array, source.Length, added.Length);
        return array;
    }
    public static T[] RemoveAll<T>(this T[] source, Predicate<T> match)
    {
        return Array.FindAll(source, a => !match(a));
    }
}
public static class ArrayHelper
{
    public static void AddRange<T>(ref T[] source, T[] added)
    {
        var size = source.Length + added.Length;
        var array = new T[size];
        Array.Copy(source, array, source.Length);
        Array.Copy(added, 0, array, source.Length, added.Length);
        source = array;
    }
    public static void RemoveAll<T>(ref T[] source, Predicate<T> match)
    {
        source = Array.FindAll(source, a => !match(a));
    }
}

注：测试方法代码以及示例源码都已经上传至代码库，有兴趣的可以看看。https://gitee.com/hugogoos/Planner