2021/6/20 刷题笔记皇位继承顺序与多叉树前序遍历，以及python defaultdict

Posted 2021-06-29 黑黑白白君

文章目录

• 皇位继承顺序
• • 【题目】
  • 【我的方法】
  • 【优化】
  • • python中的defaultdict

---

皇位继承顺序

【题目】

【我的方法】

1. 多叉树，从左到右之间是兄弟，从上到下是父子。
2. 遍历时，如果已死亡，则不放入返回列表中。

class ThroneInheritance:

    def __init__(self, kingName: str):
        self.order=[kingName]
        self.childs={kingName:[]}
        self.deaths=[]
        

    def birth(self, parentName: str, childName: str) -> None:
        if parentName in self.childs:
            self.childs[parentName].append(childName)
        else:
            self.order.append(parentName)
            self.childs[parentName]=[childName]


    def death(self, name: str) -> None:
        self.deaths.append(name)


    def getInheritanceOrder(self) -> List[str]:
        res=[]
        def func(k):
            if k not in self.deaths:
                res.append(k)
            if k in self.childs:
                for j in self.childs[k]:
                    func(j)
        if self.order:
            func(self.order[0])
        return res

# Your ThroneInheritance object will be instantiated and called as such:
# obj = ThroneInheritance(kingName)
# obj.birth(parentName,childName)
# obj.death(name)
# param_3 = obj.getInheritanceOrder()

结果超时了。。。。

【优化】

• 普通字典中调用为dict[element] = xxx，但前提是element在字典里，否则会报错。
• defaultdict的作用是在于，当字典里的key不存在但被查找时，返回的不是keyError而是一个默认值。

• python中的defaultdict

  defaultdict是Python内建dict类的一个子类。
  dict = defaultdict( factory_function )
  • 这个factory_function可以是list、set、str等等，作用是当key不存在时，返回的是factory函数的默认值，比如list对应[ ]，str对应的是空字符串，set对应set( )，int对应0。

因此可以利用defaultdict优化birth函数。

此外遍历利用递归。

class ThroneInheritance:

    def __init__(self, kingName: str):
        self.tree=defaultdict(list)
        self.deaths=set()
        self.name=kingName
        

    def birth(self, parentName: str, childName: str) -> None:
        self.tree[parentName].append(childName)            


    def death(self, name: str) -> None:
        self.deaths.add(name)


    def getInheritanceOrder(self) -> List[str]:
        ans=[]
        def func(name):
            if name not in self.deaths:
                ans.append(name)
            if name in self.tree:
                for i in self.tree[name]:
                    func(i)
        func(self.name)
        return ans

# 执行用时：496 ms, 在所有 Python3 提交中击败了82.05%的用户
# 内存消耗：63.9 MB, 在所有 Python3 提交中击败了77.78%的用户

---

【部分内容参考自】

• python中defaultdict用法详解：https://www.jianshu.com/p/bbd258f99fd3
• 皇位继承顺序：https://leetcode-cn.com/problems/throne-inheritance/solution/huang-wei-ji-cheng-shun-xu-by-leetcode-s-p6lk/