Python鸡兔同笼问题

废话少说，代码如下。 

head=int(input('一共有几个头：'))
foot=int(input('一共有几条腿：'))
flag=0      #小技巧，防止出现半只鸡、半只兔的情况
for i in range(0,head+1):
    if i*2+(head-i)*4==foot:
        print('一共有',i,'只坤')
        print('一共有',head-i,'只兔子')
        flag=1
if flag==0:#算不出来既是输入错误
    print('无解，输入有误！')