c# - 将数组中的所有非零元素拉到左侧 - C#

Question

如果我有以下数组

{ 1, 0, 0, 1, 2, 0, 1 }

我想要一个方法来获取数组并将其更改为

{ 1, 1, 2, 1, 0, 0, 0 }

执行此操作的最佳算法是什么？是否可以在 O(N) 时间内完成？

这个问题基本上是我的确切问题，除了在 python 中而不是 c# 中，以防我不清楚:(唯一的区别是将零移到右边，而不是左边) how to move all non-zero elements in a python list or numpy array to one side?

谢谢

编辑:我遇到了另一个我一开始没有考虑的问题。我实际上试图在一个二维数组上运行这个算法，但只在一个特定的维度上。我将如何更改以解决这个问题？

Answer 1

这里有一种方法可以做到这一点。

var original = new int[] { 1, 0, 0, 1, 2, 0, 1 };
var nonZeroes = original.Where(x => x != 0); //enumerate once
var numberOfZeroes = original.Count() - nonZeroes.Count(); 
return nonZeroes.Concat(Enumerable.Repeat(0, numberOfZeroes)).ToArray();