需要一种算法来优化分层形式的python词典分组

Question

之前我问了一个关于如何在层次结构的列表中对字典进行分组的问题。 Grouping Python dictionaries in hierachical form

此方法工作正常，但我现在需要优化。字典有很大的内存开销，当我们处理大量的记录时，这种分组变得非常慢。

我想知道是否有任何算法可以用来减少分组所需的时间。

P.S：我不介意改变数据结构，只要它给我相同的分层格式。

感谢。