202006-1 线性分类器

#include<iostream>
using namespace std;
//用结构体来表示每一个点
struct Node{
    int x;
    int y;
    char type;
};

int main()
{
    Node node[1005];//最多有1005个点 
    int n,m;
    cin>>n>>m;//输入n个点和m个直线的数目
	 
    for(int i=0;i<n;i++)//输入n个点的坐标及类别 
    {
        cin>>node[i].x>>node[i].y>>node[i].type;
    }
    int theta0,theta1,theta2;
    //for循环m次验证m个直线 
    for(int i=0;i<m;i++)
    {
        cin>>theta0>>theta1>>theta2;//输入直线方程的三个参数
		 
        //把点带入直线方程，如果值大于0，则记为1，小于0则记为0,
        //同类型的点带入直线方程应该具有相同的正负性，以此来判断直线是否符合标准
        
        int a[1005]={0};//每个点有一个a数组，b数组 
        int b[1005]={0};
        int a_i=0,b_i=0;
        for(int j=0;j<n;j++)
        {
            if(node[j].type=='A')//类别A 
            {
                if(theta0 + theta1 * node[j].x + theta2 * node[j].y > 0) a[a_i++] = 1;//点在直线上方 
                else a[a_i++] = 0;
            }
            else//类别B
            {
                if(theta0 + theta1 * node[j].x + theta2 * node[j].y > 0) b[b_i++] = 1;
                else b[b_i++] = 0;
            }
        }
        //标志符flag判断直线是否符合标准
        //a_i表示类别A的点的数量，1表示在直线上，0表示在直线下 
		//b_i表示类别B的点的数量 
        bool flag = true;
        for(int j=0;j<a_i-1;j++)//所有类别A的点 
        {
            if(a[j]!=a[j+1]) //同一类别点在直线上下不相等，说明直线不能分割两类别的点 
            {
                flag = false;
                break;
            }
            else continue;
        }
        for(int j=0;j<b_i-1;j++)//所有类别B的点 
        {
            if(b[j]!=b[j+1])
            {
                flag = false;
                break;
            }
            else continue;
        }
        if(flag==true) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}