Python

【题目描述】

​\ 考虑一个简单的二分类问题——将二维平面上的点分为$A $和$B$ 两类。
\​ 训练数据包含$n $个点,其中第$i $个点$(1 \le i \leq n)$可以表示为一个三元组$(x_i, y_i, type_i)$,即该点的横坐标、纵坐标和类别。
​ 在二维平面上,任意一条直线可以表示为$θ_0 + θ_1x + θ_2y = 0 $的形式,即由$θ_0、θ_1 和
θ_2 $三个参数确定该直线,且满足$θ_1、θ_2$ 不同时为$0$。
​ 基于这$n $个已知类别的点,我们想要在平面上找到一条直线作为一个线性分类器。具体来说,这条线要把训练数据中的$A、B $两类点完. 美. 分. 隔. 开来,即一侧只有$A $类点、另一侧只有$B $类点。这样,对于任意一个的未知类别的点,我们就可以根据它是位于直线的哪一侧来预测它的类别了。
​ 在本题中我们仅需要处理$m $个如下查询:给定一条直线,判断它是否能将训练数据中的$A、B$ 两类点完美分开。

【输入格式】

​ 从标准输入读入数据。
​ 输入共$n + m + 1 $行。
​ 第一行包含用空格分隔的两个正整数$n 和m$,分别表示点和查询的个数。
​ 第二行到第$n + 1 $行依次输入$n $个点的信息。第$i + 1 $行$(1 \leq i \leq n)$包含用空格分隔的三项$x_i、y_i 和type_i$,分别表示第$i $个点的横、纵坐标和类别,其中坐标为整数、类别为一个大写英文字母$A 或B$。
​ 第$n + 2$ 行到第$n + m + 1 $行依次输入$m $个查询。第$j + n + 1 $行$(1 \leq j \leq m)$包含用空格分隔的三个整数$θ_0、θ_1 和θ_2$,表示第$j$ 个查询中给定直线的三个参数。

【输出格式】

​ 输出到标准输出。
​ 输出共$m $行,每行输出一个字符串。
​ 第$j $行$(1 ≤ j ≤ m)$输出的字符串对应第$j$个查询的结果:如果给定直线可以完
美分隔$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

【样例1 解释】

​ 只有第3 个查询给出的直线能将$A、B $两类点完美分隔。

image-20200923200933463

【子任务】

  • 输入数据保证不 存 在恰好落在给定直线上的点;
  • $0 < n ≤ 10^3、0 < m ≤ 20,$且$A、B $两类点的数量均不为$0$;
  • 所有点的坐标和给定直线的三个参数均为整数,且绝对值$≤ 10^6$;
  • 任意两个点的坐标不完全相同。

image-20200923200839233

【解题程序】

#输入
n, m = map(int, input().split())
class_A = []
class_B = []
for _ in range(n):
    t = input().split()
    if t[-1] == 'A':
        class_A.append(list(map(int, t[:2])))
    else:
        class_B.append(list(map(int, t[:2])))

#判断这一类的点是否被直线完美分割
def judge_class(a, b, c, class_):
    for point in class_:
        lineValue = a + b * point[0] + c * point[1]
        pointValue = lineValue > 0
        try:
            if pointValue != preValue:
                #若现在的值与前一个值不等,则不是完美分割
                return False
        except NameError:
            #第一个点没有前一个值,会报错
            pass
        preValue = pointValue #记录前一个值
    #完美分割一个类
    return True

for _ in range(m):
    #给定直线的三个参数
    a, b, c = map(int, input().split())
    flag = 1
    for class_ in (class_A, class_B):
        judge = judge_class(a, b, c, class_)
        if not judge:
            flag = 0
            print("No")
            break
    if flag:
        print("Yes")

评论

This is just a placeholder img.