第十二届蓝桥杯初赛试题B直线

算法跑的时间有点慢，但是结果是准确的，引入set去重复会加快速度 改天有空再写个c++版本

import math
res = []
li1 = []
 
#我们不考虑平行直线 平行直线数目就等于m+n
m, n = map(int,input().split())
count = m+n
for x in range(m):
    for y in range(n):
        li1.append([x,y])
 
def sameline(res,li1):
    for a in li1:
        for b in li1:
            flag =0
            if a[0]==b[0] or a[1]==b[1]:
                continue
            else:
                k = (b[1]-a[1])/(b[0]-a[0])
                b = a[1] - k*a[0]
            for p in res:
#判断double 类型是否相等 只要在绝对值范围内相等即可 通常取1e-8
                if abs(k-p[0])<10**(-8) and abs(b-p[1])<10**(-8):
                    flag = 1
            if flag!=1:
                #print(k,b)
                res.append([k,b])
    return res
            
        
res=sameline(res,li1)
#40257
print(len(res)+count)