CCF202006-1 线性分类器

题目

线性分类器（line）
　　【题目描述】
　　考虑一个简单的二分类问题——将二维平面上的点分为A和B两类。
　　训练数据包含n个点，其中第i个点（1≤i≤n）可以表示为一个三元组（x,y,type），即该点的横坐标、纵坐标和类别。
　　在二维平面上，任意一条直线可以表示为 θ₀+θ₁x+θ₂y=0的形式，即由θ₀，θ₁，θ₂三个参数确定该直线，且满足θ₀，θ₁不同时为0。
　　基于这n个已知类别的点，我们想要在平面上找到一条直线作为一个线性分类器。具体来说，这条线要把训练数据中的A、B两类点完美分隔开来，即一侧只有A类点、另一侧只有B类点。这样，对于任意一个的未知类别的点，我们就可以根据它是位于直线的哪一侧来预测它的类别了。
　　在本题中我们仅需要处理m个如下查询：给定一条直线，判断它是否能将训练数据中的A、B两类点完美分开。
【输入格式】
　　从标准输入读入数据。输入共n+m+1行。
　　第一行包含用空格分隔的两个正整数n和m，分别表示点和查询的个数。
　　第二行到第n+1行依次输入n个点的信息。第i+1行（1≤i≤n）包含用空格分隔的三项x,y和type，分别表示第i个点的横、纵坐标和类别，其中坐标为整数、类别为一个大写英文字母A或B。
　　第n+2行到第n+m+1行依次输入m个查询。第j+n+1行（1≤j≤m）包含用空格分隔的三个整数θ₀,θ₁和θ₂，表示第j个查询中给定直线的三个参数。

【输出格式】
　　输出到标准输出。
　　输出共m行，每行输出一个字符串。
　　第j行（1≤j≤m）输出的字符串对应第个查询的结果：如果给定直线可以完美分隔A、B两类点，则输出Yes；否则输出No。
【样例1输入】
9 3
1 1 A
1 0 A
1 -1 A
2 2 B
2 3 B
0 1 A
3 1 B
1 3 B
2 0 A
0 2 -3
-3 0 2
-3 1 1

【样例1输出】

No

No

Yes

代码

def devine():
n, m = map(int, input().split()) # 读取第一行的n、m
A = []
B = []
for i in range(n): # 根据点的标签分两个列表储存
x, y, t = input().split()
x, y = int(x), int(y)
if t == ‘A’:
A.append([x, y])
else:
B.append([x, y])
res = [] # 存放结果
for i in range(m): # 对于每一条分割线
flag = True
a, b, c = map(int, input().split()) # 判断思想：给定某一类点之后该类所有点经过该线性分类后必在同一类中不然判错
neg = 0
pos = 0
for j in A:
z = j[0]*b + j[1]*c + a
if z > 0:
pos += 1
else:
neg += 1
if pos != 0 and neg != 0:
flag = False
neg = 0
pos = 0
for j in B:
z = j[0] * b + j[1] * c + a
if z > 0:
pos += 1
else:
neg += 1
if pos != 0 and neg != 0:
flag = False
if flag == False:
res.append(“No”)
else:
res.append(“Yes”)

for i in res:
    print(i)

devine()