从 HashSet 获取第一个（或任何）值

Question

目前，我的代码如下所示：

    private List<Node> dirtyNodes = new List<Node> dirtyNodes();

    public void UpdateDirtyNodes()
    {
        while(dirtyNodes.Count > 0)
        {
            Node nodeToUpdate = dirtyNodes[0];
            nodeToUpdate.UpdateNode();
            dirtyNodes.Remove(nodeToUpdate);
        }
    }

    public void MarkNodeDirty(Node node)
    {
        if(!dirtyNodes.Contains(node))
        {
            dirtyNodes.Add(node);
        }
    }

    public void MarkNodeClean(Node node)
    {
        dirtyNodes.Remove(node);
    }

这是代码中对性能至关重要的部分，它比我想要的要慢，因为dirtyNodes.Contains 在大多数情况下必须遍历整个数组。我想用HashSet 替换List，因为它应该更快，但我不知道如何使用UpdateDirtyNodes()。

困难在于UpdateNode() 可以随时从dirtyNodes 添加或删除节点，因此有点尴尬的while 循环。有没有办法从 HashSet 中获取“第一个”值？顺序无关紧要，我只需要停留在 while 循环中，直到 dirtyNodes 为空，然后更新接下来出现的任何节点。

我宁愿避免使用 Linq，因为此代码将成为库的一部分，我不想强​​迫它们包含 Linq。

我该怎么做？

Answer 1

结果证明直接使用枚举器很容易：

    public void UpdateDirtyNodes()
    {
        while(dirtyNodes.Count > 0)
        {
            using(HashSet<Node>.Enumerator enumerator = dirtyNodes.GetEnumerator())
            {
                if(enumerator.MoveNext())
                {
                    Node nodeToUpdate = enumerator.Current;
                    nodeToUpdate.UpdateNode();
                    dirtyNodes.Remove(nodeToUpdate);
                }
            }
        }
    }

    public void MarkNodeDirty(Node node)
    {
        dirtyNodes.Add(node);
    }

我最初尝试过类似的方法，但不完全了解如何手动使用枚举器，但没有奏效。

它比List 快​​得多（总帧时间快约25-50%，具体取决于仅从那一次更改的情况）所以我很高兴。 （不要担心下面屏幕截图中的 30MB 分配 - 我正在努力。）

Answer 2

在Node 类中添加一个bool dirty 字段。这是除了保留哈希集之外的。那么MarkNodeClean() 不需要从 HashSet 中删除节点，从而减少了一些 CPU 周期。

如果你觉得在Node 类中添加一个字段太“脏”（双关语），那么只需创建一个HashSet<(Node, bool)> 而不是HashSet<Node>，但是你在分配和垃圾收集方面失去了额外的性能对象，这并不理想，因为您的代码对性能至关重要。

UpdateDirtyNodes() 将一次获取一个节点，直到 HashSet 为空。取完每个节点后，它会查看布尔标志来判断该节点是否真的脏。

附言 您应该从UpdateDirtyNodes() 中删除dirtyNodes.Clear();。这是一个竞争条件。如果在 while 循环发现dirtyNodes.Count 为0 之后添加了一个节点，则dirtyNodes.Clear(); 清除该节点而不进行处理。这是一个单独的错误，与您的问题无关。