CCF-CSP线性分类器

在这里插入图片描述
思路简单暴力

n,m=input().split()
n,m=int(n),int(m)
S=[] #读入数据点和类型
for i in range(1,n+1):
    s=list(input().split())
    s[0],s[1],s[2]=int(s[0]),int(s[1]),s[2]
    S.append(s)
Z=[]#读入直线的参数
for i in range(n+1,m+n+1):
    z=list(input().split())
    z[0],z[1],z[2]=int(z[0]),int(z[1]),int(z[2])
    Z.append(z)
for i in range(len(Z)):
    type1 = set()#对m条线分别进行判断,集合可以去重
    type2 = set()
    for j in range(len(S)):
        if Z[i][0]+S[j][0]*Z[i][1]+S[j][1]*Z[i][2]>0:
            type1.add(S[j][2])
        if Z[i][0]+S[j][0]*Z[i][1]+S[j][1]*Z[i][2]<0:
            type2.add(S[j][2])
    if len(type1)==1 and len(type2)==1:
        print('Yes')
    else:
        print('No')

你可能感兴趣的:(python,算法,csp)