简单
技术面试0 次浏览

在腾讯的社交产品中,常常需要对用户的好友关系进行管理。请编写一个简单的 Python 函数,实现判断两个用户是否为好友的功能。假设好友关系存储在一个字典中,字典的键是用户 ID,值是该用户的好友 ID 列表。

腾讯算法工程师
Python数据结构社交关系

答题要点

推荐使用直接查找的答题框架。关键要点如下:1. 检查输入的两个用户 ID 是否都存在于好友关系字典中,如果有一个不存在,直接返回 False。2. 对于存在的两个用户,查看第一个用户的好友列表中是否包含第二个用户的 ID,若包含则返回 True。3. 若第一个用户的好友列表中不包含第二个用户,再查看第二个用户的好友列表中是否包含第一个用户的 ID,若包含也返回 True。示例话术:首先检查输入的两个用户 ID 是否在好友关系字典里,若都存在,就去看其中一个用户的好友列表是否有另一个用户 ID,若有则为好友关系,返回 True,若都没有则返回 False。