题目
题目描述
考虑一个简单的二分类问题——将二维平面上的点分为 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;
}