中国剩余定理用python实现

中国剩余定理

任务描述

本关任务：编写一个能计算任意组数的中国剩余定理的程序。

相关知识

为了完成本关任务，你需要掌握：1.中国剩余定理的原理

解释

在《孙子算经》中有这样一个问题：“今有物不知其数，三三数之剩二（除以3余2），五五数之剩三（除以5余3），七七数之 剩二（除以7余2），问物几何？”这个问题称为“孙子问题”，该问题的一般解法国际上称为“中国剩余定理”。 有兵一队，若列成五行纵队，则末行一人，成六行纵队，则末行五人，成七行纵队，则末行四人；成十一行纵队，则末行十人，求兵数。 射雕英雄传::黃蓉道：“也不用這般硬記， 我念一首詩給你听，那就容易記了： 三人同行七十稀，五樹梅花廿一枝，七子團圓正半月，余百零五便得知。”

编程要求

根据提示，在右侧编辑器补充代码，使用中国剩余定理，输入包含n组数据[a1,m1,a2,m2,a3,m3,…]，输出x。

测试说明

平台会对你编写的代码进行测试：

1. 测试输入： 2,3,2 3,5,7 预期输出： 23

2. 测试输入： 1,5,4,10 `5,6,7,11’

   预期输出： 2111

代码实现

# -*- coding: UTF-8 -*-
def Get_Mi(m_list, M):  # 获取所有的Mi
    M_list = []
    for mi in m_list:
        M_list.append(M // mi)
    return M_list


def Get_ei_list(M_list, m_list):  # 取所有的Mi的逆元
    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 0 == b:
        x = 1;
        y = 0;
        q = a
        return x, y, q
    xyq = Get_ei(b, a % b)
    x = xyq[0];
    y = xyq[1];
    q = xyq[2]
    temp = x;
    x = y;
    y = temp - a // b * y
    return x, y, q


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_inverse = Get_ei_list(Mi_list, m_list)
    x = 0
    for i in range(len(a_list)):  # 开始计算x
        x += Mi_list[i] * Mi_inverse[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))