CSP 202006-1 线性分类器 python实现+详解

试题

代码

# 读入m,n m, n = [int(i) for i in input().split()] # 创建字典存放已知点 dict = {} for i in range(m): line = input().split() x, y, t = line x, y = int(x), int(y) dict[(x, y)] = t # 逐条线地去判断每一个点 # 创建两个集合，一个存放坐标代入公式后大于零的点，另一个……小于…… # 如果线可以将A，B两类点分开，则每个集合只会存在A/B一种类型 for i in range(n): setSmall = set() setBig = set() line = input().split() a, b, c = [int(i) for i in line] for k, v in dict.items(): x, y = k[0], k[1] if a + b*x + c*y < 0: setSmall.add(v) else: setBig.add(v) if len(setSmall) == len(setBig) == 1: print('Yes') else: print('No')