百练 4139 不定方程求解

描述

给定正整数a，b，c。求不定方程 ax+by=c 关于未知数x和y的所有非负整数解组数。

输入

一行，包含三个正整数a，b，c，两个整数之间用单个空格隔开。每个数均不大于1000。

输出

一个整数，即不定方程的非负整数解组数。

样例输入

2 3 18

样例输出

4

错误方法

这种方法是错误的，因为当输入(1 3 5)时，y = -1/3*x + 5/3，计算机不能做分数加法，先做除法导致精度损失。
例如：(1 3 5)->1, (2 5 7)->0的正确结果应为2和1。

a, b, c = map(int, input().split())
# y = -a/b*x + c/b
# a, b, c是三个正整数，-a/b<0, c/b>0
# 从x=0开始，到y<0结束
x = 0
cnt = 0
while True:  
    y = -1 * a / b * x + c / b
    print('x = ' + repr(x) + ', y = ' + repr(y))
    if y < 0:
        break
    if y == int(y):
        print('!!')
        cnt += 1
    x += 1

print(cnt)

正确方法

做乘法不会损失精度

a, b, c = map(int, input().split())
cnt = 0
for x in range(c + 1):
    for y in range(c + 1):
        if a * x + b * y == c:
            print('x = ' + repr(x) + ', y = ' + repr(y))
            cnt += 1
print(cnt)