寻求具备确定性迭代顺序的Python set子类优化方案

你需要的是一个和Python原生set API完全一致、迭代顺序仅由集合内元素决定的极简实现，先看你给出的初始代码：

class MySet(UserDict):
 
    def __contains__(self, element):
        return element in self.data
 
    def __iter__(self):
        # from a comment by @juanpa.arrivillaga
        return iter(sorted(self.data))

    def __str__(self) -> str:
        return "{" + ", ".join([str(k) for k in self._keys()]) + "}"
 
    def add(self, element):
        # arbitraily map all elements to True
        self.data[element] = True
 
    def _keys(self):
        return sorted(self.data.keys())

这个实现可以从几个方向优化，让它更简洁、高效且贴合set的规范：

优化点1：换用更合适的父类 #

直接继承collections.abc.MutableSet而非UserDict。MutableSet是Python为集合类定义的抽象基类，已经帮你实现了大部分set的默认方法（比如update、union、difference等），你只需要实现几个核心方法即可，不用重复造轮子，代码量会大幅减少，也更符合set的接口规范。

优化点2：避免重复排序的性能损耗 #

你的代码里__iter__和_keys每次调用都会对data.keys()重新排序，当集合元素较多时，反复排序会浪费性能。可以维护一个缓存的有序键列表，只在元素添加/删除时重新排序，迭代和字符串展示直接用这个缓存列表，减少重复计算。

优化点3：简化冗余方法 #

_keys方法属于冗余封装，直接用缓存的有序列表就能替代，没必要单独抽出来。另外__str__里的列表推导式可以换成map(str, ...)，写法更简洁。

优化后的完整实现 #

from collections.abc import MutableSet

class OrderedSet(MutableSet):
    def __init__(self, iterable=None):
        self._data = dict()  # 用dict存元素，保证唯一性和可哈希性
        self._ordered_keys = []  # 缓存有序的元素列表
        if iterable is not None:
            self.update(iterable)
    
    def __contains__(self, element):
        return element in self._data
    
    def __iter__(self):
        return iter(self._ordered_keys)
    
    def __len__(self):
        return len(self._data)
    
    def __str__(self):
        return "{" + ", ".join(map(str, self._ordered_keys)) + "}"
    
    def add(self, element):
        if element not in self._data:
            self._data[element] = True
            self._ordered_keys.append(element)
            self._ordered_keys.sort()  # 新增元素后重新排序，保证确定性顺序
    
    def discard(self, element):
        if element in self._data:
            del self._data[element]
            self._ordered_keys.remove(element)

这个实现的优势：

• 完全贴合Python原生set的API，继承MutableSet后自动拥有update、union、intersection等所有set方法
• 仅在元素增删时排序，迭代和字符串展示直接用缓存的有序列表，性能更优
• 代码更简洁，逻辑清晰

内容的提问来源于stack exchange，提问作者OrenIshShalom