python数据结构与算法练习-（ECNU上机）

解方程

链接: link.

简而言之，本题任务就是解方程。共有两个子任务。

子任务 1：小学生
作为小学生，我们只会解一元一次方程，一元一次方程最终都可以化为ax = n 的形式。现在问：对于给定的x，要使得x有正整数解，总共可以取多少个不同的 a 呢？

子任务 2：中学生
作为中学生，我们只会解二元一次不定方程，二元一次不定方程最终都可以化为 ax+by=n 的形式。现在问：对于给定的n，要使得 x,y 有正整数解，总共可以取多少对不同的 (a,b) 呢？

输入格式
输出一行两个整数 q,n(q属于(1,2)) 。

q=1 表示你现在要解决小学生的情况，q=2 表示你现在要解决中学生的情况。

数据规模约定（每个测试点占本题总分值的10%）：

输出格式
输出一个整数，表示答案。

样例

#input
2 4
#output
6

#input
1 10
#output
4

提示

思路：

二元一次方程ax=by=c有正整数解的条件是a、b的最大公约数能整除c。
本题我并没有完全做出来，后面时间复杂度超了，在此记录一下。

python实现

#时间复杂度超出范围了
q,n=map(int,input().split())

if q == 1:
    if n== 1:
        print(1)
    else:
        res = 0
        for i in range(1,n+1):
            if n %i == 0:
                res += 1
        print(res)
#二元一次方程ax=by=c有正整数解的条件是a、b的最大公约数能整除c
else:

    if n == 1:
        print(0)
    else:
        sum = 0
        res = []

        for i in range(1,n+1):
            for j in range(1,n-i+1):
                m_res = []
                # if i+j<=n:
                min_m = min(i,j)
                for x in range(1,min_m+1):
                    if i%x==0 and j%x==0:
                        m_res.append(x)
                #求出最大公约数之后，还要判断对于此时的i,j是否可以获得正整数解
                if n % max(m_res) == 0:
                    for b in range(1,n+1):
                        if b*i < n:
                            if (n- b*i)%j == 0 :
                                res.append((i,j))
        print(len(set(res)))

仅记录刷题过程以及需要注意的知识点，方便自己复习。