202006-1 CCF认证考试Python版线性分类器

我的思路是当A类大于0添加1，小于0添加0，B类反之，之后根据数组a里面是否为一种元素判断。100分

n,m=input().split()
n,m=int(n),int(m)
X=[]
Y=[]
T=[]
O_1=[]
O_2=[]
O_3=[]
r=[]
re=8
for i in range(n):
	x,y,t=input().split()
	x,y=int(x),int(y)
	t=str(t)
	X.append(x)
	Y.append(y)
	T.append(t)
for j in range(m):
	O_4,O_5,O_6=input().split()
	O_4,O_5,O_6=int(O_4),int(O_5),int(O_6)
	O_1.append(O_4)
	O_2.append(O_5)
	O_3.append(O_6)
for o in range(m):
	a=[]
	for k in range(n):
		if T[k]=="A":
			y=O_1[o]+O_2[o]*X[k]+O_3[o]*Y[k]
			if y>0:
				a.append(1)
			else:
				a.append(0)
		if T[k]=="B":
			y=O_1[o]+O_2[o]*X[k]+O_3[o]*Y[k]
			if y>0:
				a.append(0)
			else:
				a.append(1)
	if (1 in a) and (0 in a):
		re=1
	else:
		re=0
	if re==1:
		r.append("No")
	else:
		r.append("Yes")
for q in range(m):
	print(r[q])