【每周一坑】鸡兔同笼

cczztt

1. from sympy import symbols, solve
   
2. from math import modf
   
3. 
   
4. def isInt(inp):
   
5.     while True:
   
6.         try:
   
7.             inp = float(inp)
   
8.             dec = modf(inp)
   
9.             if dec[0] == 0:
   
10.                 return inp
    
11.                 break
    
12.             else:
    
13.                 inp = input("Input Error! Try Again: ")
    
14.         except:
    
15.             inp = input("Input Error! Try again: ")
    
16. 
    
17. head = input("Input heads: ")
    
18. head = isInt(head)
    
19. foot = input("Input foot: ")
    
20. foot = isInt(foot)
    
21. 
    
22. result = {}
    
23. x = symbols('x')
    
24. y = symbols('y')
    
25. result = solve([x + y - head, 2 * x + 4 * y - foot], [x, y])
    
26. print(result)
    
27. if result[x] >= 0 and result[y] >= 0 and modf(result[x])[0] + modf(result[y])[0] == 0:
    
28.     print("Chicken: %d, rabbits: %d."%(result[x], result[y]))
    
29. else:
    
30.     print("Unsolvable!")

复制代码