百钱白鸡问题

#! /usr/bin/python3
# -*- coding: utf-8 -*-

def buy_chicken():
  '''鸡翁一值钱五，鸡母一值钱三，鸡雏三值钱一。百钱买百鸡，问鸡翁、鸡母、鸡雏各几何？
    Args: 

    Returns: 鸡翁、鸡母、鸡雏个数
  '''
  results = []
  for i in range(0, 20):
    res = []
    for j in range(0, 34):  # 5*i + 3*j + (100-i-j)/3 = 100
      if (100 - i - j) % 3 != 0:
        continue
      if (5*i + 3*j + (100-i-j)//3 == 100):
        res.extend([i, j, 100-i-j])
    if len(res) > 0:
      results.append(res)
  print(results)

def buy_chicken_2():
  '''鸡翁一值钱五，鸡母一值钱三，鸡雏三值钱一。百钱买百鸡，问鸡翁、鸡母、鸡雏各几何？
    Args: 

    Returns: 鸡翁、鸡母、鸡雏个数
  '''
  results = []
  for i in range(0, 15):
    if ((300 + 3*i) % 4) == 0 and ((100 - 7*i) % 4) == 0:
      results.append([i, int((100 - 7*i)/4), int((300 + 3*i)/4)])
  print(results)


buy_chicken()
buy_chicken_2()       
# 运行结果：
# [[0, 25, 75], [4, 18, 78], [8, 11, 81], [12, 4, 84]]
# [[0, 25, 75], [4, 18, 78], [8, 11, 81], [12, 4, 84]]

函数buy_chicken()的复杂度为O(n*n)，进行优化后buy_chicken_2()算法复杂度为O(n)；

优化过程：
设鸡翁、鸡母及鸡雏的个数分别为x、y、z，通过以下表达式

$$\begin{gathered} x+y+z=100 \\ 5x+3y+z/3=100 \end{gathered}$$
可以推到出
$$\begin{gathered} y=(100-7x)/4 \\ z=(300+3x)/4 \end{gathered}$$
而根据条件x<20, y>0得出x<15，从而将复杂度降低为O(n)