在《孙子算经》中有这样一个问题:“今有物不知其数,三三数之剩二(除以3余2),五五数之剩三(除以5余3),七七数之 剩二(除以7余2),问物几何?”这个问题称为“孙子问题”,该问题的一般解法国际上称为“中国剩余定理”。
有兵一队,若列成五行纵队,则末行一人,成六行纵队,则末行五人,成七行纵队,则末行四人;成十一行纵队,则末行十人,求兵数。
射雕英雄传::黃蓉道:“也不用這般硬記, 我念一首詩給你听,那就容易記了: 三人同行七十稀,五樹梅花廿一枝,七子團圓正半月,余百零五便得知。”
中国剩余定理计算公式:
M=m1m2…*mk
# -*- coding: UTF-8 -*-
#获取所有的Mi
def Get_Mi(m_list, M):
M_list = []
for mi in m_list:
M_list.append(M // mi)
return M_list
# 取所有的Mi的逆元
def Get_ei_list(M_list, m_list):
ei_list = []
for i in range(len(M_list)):
ei_list.append(Get_ei(M_list[i], m_list[i])[0])
return ei_list
#利用扩展欧几里得算法计算逆元
def Get_ei(a, b):
# 计算逆元ei
if a < b:
t = a
a = b
b = t
x1, x2, x3 = 1, 0, a
y1, y2, y3 = 0, 1, b
while True:
if y3 == 0:
return x1,x2,x3
break
elif y3 == 1:
return y1,y2,y3
break
else:
Q = x3 // y3
t1, t2, t3 = x1 - Q * y1, x2 - Q * y2, x3 - Q * y3
x1, x2, x3 = y1, y2, y3
y1, y2, y3 = t1, t2, t3
def crt(a_list, m_list):
# 计算中国剩余定理,返回计算结果
M = 1 # M是所有mi的乘积
for mi in m_list:
M *= mi
Mi_list = Get_Mi(m_list, M)
Mi_reverse = Get_ei_list(Mi_list, m_list)
x = 0
for i in range(len(a_list)): # 开始计算x
x += Mi_list[i] * Mi_reverse[i] * a_list[i]
x %= M
return x
if __name__ == '__main__':
a_list = list(map(int, input().split(",")))
m_list = list(map(int, input().split(",")))
print(crt(a_list, m_list))
测试输入:
2,3,2
3,5,7
预期输出:
23
测试输入:
1,5,4,10
5,6,7,11
预期输出:
2111