python - 非递归函数 follow_me(d, s) 其中 d 是字典，s 是字符串

Question

我目前正在备考并且一直在努力解决这个问题，我最大的问题似乎是正确地遍历字典。问题本身是一个非递归函数 follow_me(d, s) 其中 d 是字典，s 是字符串。该字符串可能是字典的关键字。与该键关联的值可能反过来是字典的另一个键。继续查找键，直到找到没有关联值的键。然后，返回该 key 。

我试过只遍历键然后遍历值，但是，似乎最理想的遍历方法是在 d.items() 中使用 k,v。我对 python 很陌生，这可能并不理想，因为我可能缺乏一些理解，这可能是我无法找到解决方案的最大问题。

def follow_me(d, s):
    for k,v in d.items():
        try:
            if s == k:
                s = v
                continue

            else:
                return s

        except:
            break

d = {'badger':'doe', 'doe':'fox', 'fox':'hen','hen':'flea',
'sparrow':'spider', 'zebra':'lion', 'lion':'zebra'}
print(follow_me(d, 'badger'))
print(follow_me(d, 'fox'))
print(follow_me(d, 'sparrow'))
print(follow_me(d, 'zebra'))
print(follow_me(d, 'aardvark'))

预期结果:跳蚤跳蚤蜘蛛错误的土豚

实际结果:跳蚤狐狸麻雀斑马土豚

Zebra 应该是错误的，因为它在 zebra 和 lion 之间无限循环。任何帮助/提示将不胜感激。

Answer 1

你很接近，但你不需要遍历键；一个简单的 while 循环就可以了。

def follow_me(d, s):
    while s in d:
        s = d[s]
    return s

也就是说，您应该通过将检查的值添加到 set 来检查循环:

def follow_me(d, s):
    checked_keys = set()
    while s in d:
        checked_keys.add(s)
        s = d[s]
        if s in checked_keys:
            raise Exception('Cycle detected')
    return s