老师让用python编写一个输入省份列出对应城市的程序，这个代码求大神指点一下哪里不对

shengfen=["河北","江苏","山东"]
chengshi=[["石家庄","邢台","保定"],["苏州","南昌","扬州"],["菏泽","济南","青岛"]]

def search(c):
for i in range(len(shengfen)):
for x in shengfen[i]:
if x==c:
print(c,"省有如下城市：",chengshi[i])
return
print("省份没有城市存档")

c=input("输入省名：")
search(c)

shengfen = ["河北", "江苏", "山东"]
chengshi = [["石家庄", "邢台", "保定"], ["苏州", "南昌", "扬州"], ["菏泽", "济南", "青岛"]]


def search(c):
    for i in range(len(shengfen)):
        if shengfen[i] == c:
            print(c, "省有如下城市：", chengshi[i])
            return
    print("省份没有城市存档")


c = input("输入省名：")
search(c)