CCF-CSP 202006-1 线性分类器

问题描述

试题编号：	202006-1
试题名称：	线性分类器
时间限制：	1.0s
内存限制：	512.0MB
问题描述

思路：

读完题慢慢来就行，根据数学知识，在一条直线上或下，带入直线一般式符号相反，而这里恰好有A和B两种，也就是说，如果A和B我将其输入为+1和-1表示，如果能完美分割，那么 一般式*（1或-1）最后结果应该都是同一个符号。比如我将A设置为1，将B设为-1，那么如果带入A的点的一般式*1>0(全部）那么就说明A类点在同一侧，而同时如果带入B的点的一般式*（-1）>0(全部）那就说明B类点全在另一侧，也就可以说明这个线完美分割了AB两类点。

需要注意的是这里的一般式顺序有点颠倒，是（C+Bx+Cy)

代码如下：

#include<bits/stdc++.h>
using namespace std;
bool check(int a,int b,int c, vector<int>& x, vector<int>& y, vector<int>& t,int n){
    
    
    int cnt1=0;
    int cnt2=0;
    for(int i=0;i<n;i++){
        if((a+b*x[i]+c*y[i])*t[i]>0){
            cnt1++;
        }
        else if((a+b*x[i]+c*y[i])*t[i]<0){
            cnt2++;
        }
    }
    //cout<<"cnt1="<<cnt1<<endl;
    //cout<<"cnt2="<<cnt2<<endl;
    if(cnt1==n||cnt2==n){
        return true;
    }
    else{
        return false;
    }
    
}
int main(){
    int n,m;cin>>n>>m;
    vector<int>x(n),y(n),t(n);
    char type;//临时存储t[i];
    for(int i=0;i<n;i++){
        cin>>x[i]>>y[i];
        cin>>type;
        if(type=='A'){
            t[i]=1;
        }
        else if(type=='B'){
            t[i]=-1;
        }
    }
    int n1,n2,n3;
    for(int j=0;j<m;j++){
        cin>>n1>>n2>>n3;
        if(check(n1,n2,n3,x,y,t,n)){
            cout<<"Yes"<<endl;
        }
        else{
            cout<<"No"<<endl;
        }
        
    }


}