「csp模拟试题202006-1」线性分类器 C/C++ 100分

目录

• 题目
• 输入输出样例
• 题解思路
• 题解源码（c/c++）

题目 ：线性分类器

【题目描述】

考虑一个简单的二分类问题——将二维平面上的点分为 A 和 B 两类。

训练数据包含 n 个点，其中第 i 个点（1≤i≤n）可以表示为一个三元组 (xi,yi,typei)， 即该点的横坐标、纵坐标和类别。

在二维平面上，任意一条直线可以表示为 θ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≤i≤n）包含用空格分 隔的三项 xi、yi 和 typei，分别表示第 i 个点的横、纵坐标和类别，其中坐标为整数、类 别为一个大写英文字母 A 或 B。

第 n + 2 行到第 n + m + 1 行依次输入 m 个查询。第 j + n + 1 行（1≤ j≤m）包含 用空格分隔的三个整数 θ0、θ1 和 θ2，表示第 j 个查询中给定直线的三个参数。

【输出格式】

输出到标准输出。

输出共 m 行，每行输出一个字符串。

第 j 行（1 ≤ j ≤ m）输出的字符串对应第 j 个查询的结果：如果给定直线可以完 美分隔 A、B 两类点，则输出 Yes；否则输出 No。

【子任务】

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

输入输出样例

• 输入样例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 两类点完美分隔

题解思路

题目已经说了，不存在刚好落在直线上的点。而且 AB 两类的点的个数均不为 0，那么我只需要分类两类点在直线上面和下面即可（当直线斜率为 90° 时，左边当做下面，右边当做上面。）。上面的点返回 1，下面的点返回-1。统计 flag 的绝对值，若绝对值等于 n 值，那么就说明直线完美分割开两类点。

题解（c/c++）

#include<bits/stdc++.h>
using namespace std;
int isUpOrDown(int a,int b,int c,int x,int y)
{
	/* 直线的上面和下面：up返回1  down返回-1 */
	if (a + b * x + c * y > 0)
	{
		return 1;
	}
	else
	{
		return -1;
	}
}
struct P
{
	int x;
	int y;
	char type;
};
struct Line
{
	/*  a+bx+cy=0  */
	int a;
	int b;
	int c;
};
int main(void)
{
	int n, m;	//点、查询个数
	struct P point[1010];
	struct Line num[21];
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		cin >> point[i].x >> point[i].y >> point[i].type;
	}
	for (int i = 0; i < m; i++) {
		int flag=0;
		cin >> num[i].a >> num[i].b >> num[i].c;
		for (int j = 0; j < n; j++) {
			if (point[j].type == 'A') {
				flag+=isUpOrDown(num[i].a, num[i].b, num[i].c, point[j].x, point[j].y);
			}
			else {
				flag-= isUpOrDown(num[i].a, num[i].b, num[i].c, point[j].x, point[j].y);
			}
		}
		if (abs(flag)==n) {
			cout << "Yes" << endl;
		}
		else {
			cout << "No" << endl;
		}
	}
	return 0;
}