CCF-202006-1---线性分类器

CCF-202006-1—线性分类器

题目：

代码：
第一版

#include <iostream>
using namespace std;

int main(){
	int n,m;
	cin>>n>>m;
	int A[n][2]; //点的坐标 
	char T[n]; //点的类型 
	int a,b,c; //直线的系数 
	int k=0,count=0; //k是循环次数，count记录分类正确的次数 
	char f[2]; //f下标为0，记录小于0的情况下的类型；下标为1，记录大于0的情况下的类型 
	for(int i=0;i<n;i++){
		cin>>A[i][0]>>A[i][1]>>T[i];
	}
	while(k<m){
		count=0;
		cin>>a>>b>>c;
		if((a+b*A[0][0]+c*A[0][1])>0){
			if((T[0]=='A')){
				f[0]='B';
			}else{
				f[0]='A';
			}
			f[1]=T[0];
		}else if((a+b*A[0][0]+c*A[0][1])<0){
			if((T[0]=='A')){
				f[1]='B';
			}else{
				f[1]='A';
			}
			f[0]=T[0];
		}
		for(int i=1;i<n;i++){
			if((a+b*A[i][0]+c*A[i][1])>0){
				if(T[i]!=f[1]){
					cout<<"No"<<endl;
					break;
				}else{
					count++;
				}
			}else if((a+b*A[i][0]+c*A[i][1])<0){
				if(T[i]!=f[0]){
					cout<<"No"<<endl;
					break;
				}else{
					count++;
				}
			}
		}
		if(count==n-1){
			cout<<"Yes"<<endl;
		}
		k++;
	}
	return 0;
}

第二版

#include <iostream>
using namespace std;

int main(){
	int n,m;
	cin>>n>>m;
	char flag1,flag2; //flag1记录大于0的情况 ，flag2记录小于0的情况  
	int d[n][2];
	int f[m][3];
	char t[n];
	int x,y,z;
	char type;
	for(int i=0;i<n;i++){
		cin>>x>>y>>type;
		d[i][0]=x;
		d[i][1]=y;
		t[i]=type;
	}
	for(int i=0;i<m;i++){
		cin>>x>>y>>z;
		f[i][0]=x;
		f[i][1]=y;
		f[i][2]=z;
	}
	int count;
	for(int i=0;i<m;i++){
		count=0;
		for(int j=0;j<n;j++){
			if(f[i][0]+f[i][1]*d[0][0]+f[i][2]*d[0][1]>0){
				if(t[0]=='A'){
					flag1='A';
					flag2='B';
				}else{
					flag1='B';
					flag2='A';
				}
			}else{
				if(t[0]=='A'){
					flag2='A';
					flag1='B';
				}else{
					flag2='B';
					flag1='A';
				}
			}
			if(f[i][0]+f[i][1]*d[j][0]+f[i][2]*d[j][1]>0 && t[j]==flag1){
				count++;
			}else if(f[i][0]+f[i][1]*d[j][0]+f[i][2]*d[j][1]<0 && t[j]==flag2){
				count++;
			}
		}
		if(count==n){
			cout<<"Yes"<<endl;
		}else{
			cout<<"No"<<endl;
		}
	}
	return 0;
}