CSP 202006-1 线性分类器 python

题目描述

思路

简单来说，就是判断是否能够二分类两条直线，一般来说，如果在直线两侧，分别输入两类，一个是大于0，一个是小于0，有个符号的变换

在这里，如果每次都要判断两次符号，分别计算就比较麻烦，我们可以设置一个标志，就是说如果A点是在直线上方，那么带入就是大于0，A的标志为1，B的标志就为-1，这样子就是得到直线f乘上标志位flag都是大于0的，如果每次都大于0，就说明我们是分对了。

但是我万万没想到，这样只能拿60分，因为还有一个任务，我们要判断一下，A和B是不是都出现过，如果只有一类样本，我们也是输出No

所以最后我们只需要判断f*flag < 0，小于说明没分好类，直接就break退出了

代码

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
# @File    :   202006-1.py
# @Time    :   2021/11/20 12:24:38
# @Author  :   DKJ
# @Contact :   1016617094@qq.com
# @Software:   VScode

n,m = map(int,input().split(' '))
data = []
for i in range(n):
    # 将输入数据压入data
    data.append(list(input().split(' ')))
    
def f(z,x,y,X,Y):
    # 得到直线方程
    return z + x*X + y*Y

for i in range(m):
    key = {}
    count = {}
    z,x,y = map(int,input().split())
    
    if f(z,x,y,int(data[0][0]),int(data[0][1])) > 0:
        key['A'] = 1
        key['B'] = -1
    else:
        key['A'] = -1
        key['B'] = 1
    flag = 1
    for j in range(1,n):
        # 如果是正确分类的话，满足f*key > 0,key是进行符号变换的
        if f(z,x,y,int(data[j][0]),int(data[j][1])) * key[data[j][2]] < 0:
            flag = 0
            break
        else:
            count[key[data[j][2]]] = 1
    # 子任务，A和B都要存在，如果不存在也是No
    if 'A' and 'B' not in count.keys():
        flag = 0
    if flag:
        print('Yes')
    else:
        print('No')