三级菜单1

menu = {
    '北京': {
        '海淀': {
            '五道口': {
                'soho': {},
                '网易': {},
                'google': {}
            },
            '中关村': {
                '爱奇艺': {},
                '汽车之家': {},
                'youku': {},
            },
            '上地': {
                '百度': {},
            },
        },
        '昌平': {
            '沙河': {
                '老男孩': {},
                '北航': {},
            },
            '天通苑': {},
            '回龙观': {},
        },
        '朝阳': {},
        '东城': {},
    },
    '上海': {
        '闵行': {
            "人民广场": {
                '炸鸡店': {}
            }
        },
        '闸北': {
            '火车战': {
                '携程': {}
            }
        },
        '浦东': {},
    },
    '山东': {},
}
# 请实现一下功能：
# 显示【北京、上海、山东】，用户输入选择
# 若输入【北京】，则显示【海淀、昌平、朝阳、东城】，用户可退出、返回上一层，也可输入选择
# 若输入【海淀】，则显示【五道口、中关村、上地】，用户可退出、返回上一层，也可输入选择
# 若输入【五道口】，则显示【Sohu、网易、google】，用户可退出、返回上一层，也可输入选择

# 递归思想
def menu_func(menu):
    while True:
        for key in menu:
            print(key)
        inp = input('>>>').strip()
        if inp.upper() == 'Q': return 'q'
        if inp.upper() == 'B': return 'b'
        elif menu.get(inp):
            flag = menu_func(menu[inp])
            if flag == 'q': return 'q'
menu_func(menu)
print('已退出')

# 堆栈思想
lst = [menu]
while lst:
    for key in lst[-1]:
        print(key)
    inp = input('>>>')   
    if inp.upper() == 'Q':
        break
    elif inp.upper() == 'B':  # 返回上一层
        lst.pop()   # pop默认删除列表的最后一个元素
    elif lst[-1].get(inp):
        lst.append(lst[-1][inp])