CCF-CSP 2020.6.1 线性分类器(c++ vector)

题目

题目描述

  考虑一个简单的二分类问题——将二维平面上的点分为 A 和 B 类。
  训练数据包含 n 个点，其中第 i 个点（1 <= i <= n）可以表示为一个三元组（x, y, type），即该点的横坐标，纵坐标，和类别。
  在二维平面上，任意一条直线可以表示为Θ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。

输入与输出样例

输入

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

输出

No
No
Yes

大体思路

首先将所有点的信息储存在vector中，再循环输入直线的信息，载利用循环将每个点带入直线的信息中，利用一个变量count初始为0，若类型A且代入大于0，count++，若小于0，count–；若类型B且带入大于0，count–，若小于0，count++。最终中间count是否为点的数量m或-m输出Yes或者No

代码

#include<iostream>
#include<math.h>
#include<vector>

using namespace std;

int main()
{
	int m,n;
	cin>>m>>n;
	vector<int> a,b;
	vector<char> c;
	for(int i = 0 ; i < m ; i++)//利用vector储存点的信息 
	{
		int a1,a2;
		char a3;
		cin>>a1>>a2>>a3;
		a.push_back(a1);
		b.push_back(a2);
		c.push_back(a3);
	}
	for(int i = 0 ; i < n ; i++)//循环判断每一条直线 
	{
		int b1,b2,b3,count = 0;
		cin>>b1>>b2>>b3;
		for(int j = 0 ; j < m ; j++)//循环遍历每一个点 
		{
			if(c[j] == 'A' && b1 + b2 * a[j] + b3 * b[j] > 0)
			count ++;
			else if(c[j] == 'A' && b1 + b2 * a[j] + b3 * b[j] < 0)
			count --;
			else if(c[j] == 'B' && b1 + b2 * a[j] + b3 * b[j] > 0)
			count --;
			else if(c[j] == 'B' && b1 + b2 * a[j] + b3 * b[j] < 0)
			count ++;
		}
		if(count == m || count == (-m))//通过结果判断点是否合适 
		cout<<"Yes"<<endl;
		else
		cout<<"No"<<endl;
	}
	return 0;
}