#! /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,通过以下表达式
x
+
y
+
z
=
100
5
x
+
3
y
+
z
/
3
=
100
x + y + z = 100 \\ 5x + 3y + z/3 = 100
x+y+z=1005x+3y+z/3=100
可以推到出
y
=
(
100
−
7
x
)
/
4
z
=
(
300
+
3
x
)
/
4
y = (100 - 7x) / 4 \\ z = (300 + 3x) / 4
y=(100−7x)/4z=(300+3x)/4
而根据条件x<20, y>0得出x<15,从而将复杂度降低为O(n)