抓交通肇事犯(python)

一辆卡车违反交通规则，撞人后逃跑。现场有三人目击该事件，但都没有记住车号，只记住了车号的一些特征。甲说牌照的前两位数字是相同的；乙说牌照的后两位数字是相同的，但与前两位相同；丙是数学家，他说4位的车号刚好是一个整数的平方，请根据以上线索求出车号。

if __name__ == "__main__":
    for i in range(10):
        for j in range(10):
            if i != j:
                k = 1000*i+100*i+10*j+j
                for temp in range(31, 100):
                    if temp*temp == k:
                        print("车牌号为：", k)

引入判断标志，减少循环次数

if __name__ == "__main__":
    flag = 0
    for i in range(10):
        if flag == 1:
            break
        for j in range(10):
            if i != j:
                k = 1000*i+100*i+10*j+j
                for temp in range(31, 100):
                    if temp*temp == k:
                        print("车牌号为：", k)
                        flag = 1
                        break