线性分类器
来源:CCF
标签:
参考资料:
相似题目:
考虑一个简单的二分类问题——将二维平面上的点分为A 和B 两类。
训练数据包含 n 个点,其中第 i 个点(1 ≤ i ≤ n)可以表示为一个三元组 (xi
xi, yiyi, typeitypei),即该点的横坐标、纵坐标和类别。
在二维平面上,任意一条直线可以表示为θ0+θ1x+θ2y=0θ0+θ1x+θ2y=0的形式,即由 θ0θ0、θ1θ1 和 θ2θ2 三个参数确定该直线,且满足 θ1θ1、θ2
θ2 不同时为 0。
基于这 n 个已知类别的点,我们想要在平面上找到一条直线作为一个线性分类器。
具体来说,这条线要把训练数据中的A、B 两类点完. 美. 分. 隔. 开来,即一侧只有A 类点、另一侧只有B 类点。这样,对于任意一个的未知类别的点,我们就可以根据它是位于直线的哪一侧来预测它的类别了。
在本题中我们仅需要处理 m 个如下查询:给定一条直线,判断它是否能将训练数据中的A、B 两类点完美分开。
从标准输入读入数据。
输入共 n + m + 1 行。
第一行包含用空格分隔的两个正整数 n 和 m,分别表示点和查询的个数。
第二行到第 n + 1 行依次输入 n 个点的信息。第 i + 1 行(1 ≤ i ≤ n)包含用空格分隔的三项 xi
xi、yiyi 和 typeitypei,分别表示第 i 个点的横、纵坐标和类别,其中坐标为整数、类别为一个大写英文字母A 或B。 第 n + 2 行到第 n + m + 1 行依次输入 m 个查询。第 j + n + 1 行(1 ≤ j ≤ m)包含用空格分隔的三个整数 θ0θ0、θ1θ1 和 θ2
θ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
只有第 3 个查询给出的直线能将A、B 两类点完美分隔。
将点的坐标代入直线方程,通过计算结果大于0还是小于0,判断点在直线的哪一侧。
#include<iostream> #include<stack> #include<vector> #include<algorithm> #include<map> #include<stdio.h> #include<string.h> using namespace std; int main() { int i,n,m,x,y,a,b,c,sum,fA,fB,flag; char str; cin>>n>>m; map<pair<int,int>,char>mp; map<pair<int,int>,char>::iterator it; for(i=0;i<n;i++) { cin>>x>>y>>str; mp[make_pair(x,y)]=str; } for(i=0;i<m;i++) { flag=0;fA=0;fB=0; cin>>c>>a>>b; for(it=mp.begin();it!=mp.end();it++) { sum=c+a*it->first.first+b*it->first.second; if(it==mp.begin())//以第一个元素确定fA,fB的值(-1为<0,1为>0) { if(it->second=='A') if(sum>0)fA=1,fB=-1; else fA=-1,fB=1; else { if(sum>0)fB=1,fA=-1; else fB=-1,fA=1; } } else { //当同类型的点不在同一侧时,flag置为1. if(it->second=='A'&&fA==1&&sum<0) flag=1; else if(it->second=='A'&&fA==-1&&sum>0) flag=1; else if(it->second=='B'&&fB==-1&&sum>0) flag=1; else if(it->second=='B'&&fB==1&&sum<0) flag=1; } } if(!flag)cout<<"Yes\n"; else cout<<"No\n"; } return 0; }