第 19 次 CCF CSP 认证 202006-1 线性分类器(C++)(代码最简版)

第 19 次 CCF CSP 认证 202006-1 线性分类器(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。

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

【code】

#include<iostream>
using namespace std;

int main() {
	int n,m;
	cin>>n>>m;
	int x[n],y[n],k[3];           //k[0,1,2]分别存 θ(1,2,3)
	char c[n];                    //c[]存A/B
	for(int i=0; i<n; ++i)        //存点的数据
		cin>>x[i]>>y[i]>>c[i];
	for(int i=0; i<m; ++i) {      //m组线的数据，输入一组，出一个结果
		cin>>k[0]>>k[1]>>k[2];    // 分别存 θ(1,2,3)
		int flag=0;               //统计标记
		for(int j=0; j<n; ++j) {
			float line=0.0;       //直线y会出现小数
			line=(1.0*(k[0]+k[1]*x[j])/(-k[2]));//θ0 + θ1 x + θ2 y=0变形得到直线y的值
			if(c[j]=='A') {       //分类A/B组
				if(y[j]>line)     //y[]是点的y，如果A组的点在直线的上方，则y[j]>line
					flag+=1;      //标记+1，此时B组就是在下方,y[j]<line. 标记flag也要+1.
				else              //因为这样就是被直线分开了。（x相同时，看y的大小 ）
					flag-=1;      //相反，如果一开始A组点，在直线的下方，应该执行else语句，
				}                 // 则标记flag-1,所以B组就在上方，同样执行else语句flag-1
			else {                // 总之，如果AB全部被线分开，则最后的flag值肯定是n或者-n
				if(y[j]<line)
					flag+=1;
				else
					flag-=1;
				}
			}
		if(flag==n||flag==0-n)
			cout<<"Yes"<<endl;    //这里是“Yes”不是“YES”
		else
			cout<<"No"<<endl;
		}
	return 0;
	}